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