numpy: get random set of rows from 2d array,"A[(np.random.randint(A.shape[0], size=2)), :]"