NumPy多次元配列のi番目の列にアクセスするにはどうすればいいですか? 質問する

NumPy多次元配列のi番目の列にアクセスするにはどうすればいいですか? 質問する

与えられた条件:

test = np.array([[1, 2], [3, 4], [5, 6]])

test[i]i 番目の行を返します(例)。i番目の[1, 2]列にアクセスするにはどうすればよいでしょうか? (例)。また、これはコストのかかる操作でしょうか?[1, 3, 5]

ベストアンサー1

と:

test = np.array([[1, 2], [3, 4], [5, 6]])

列 0 にアクセスするには:

>>> test[:, 0]
array([1, 3, 5])

行 0 にアクセスするには:

>>> test[0, :]
array([1, 2])

これについては、NumPyリファレンス少なくとも私の経験では、これは速いです。ループ内の各要素にアクセスするよりもはるかに速いのは確かです。

おすすめ記事