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