Numpy: Get random set of rows from 2D array,"A[(np.random.randint(A.shape[0], size=2)), :]"