[エラー] TypeError: only integer arrays with one element can be converted to an index

エラー

行列からランダムに行を取り出そうとしたらエラーが出た。

>>> x = [[0,0],[1,1],[2,2],[3,3],[4,4],[5,5]]
>>> perm = np.random.permutation(6)[:3]
>>> perm
array([0, 2, 4])
>>> x[perm]
TypeError: only integer arrays with one element can be converted to an index

解決

行列をnp.arrayにしたら解決した

>>> x = np.array(x)
>>> x[perm]
array([[0, 0],
       [2, 2],
       [4, 4]])