1. ホーム
  2. python

[解決済み] NumPy多次元配列のi番目の列にアクセスする方法は?

2022-03-17 03:02:52

質問

持っているとします。

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

test[i] ゲットミー 配列の行 (例 [1, 2] ). どのようにすれば の列がありますか?(例 [1, 3, 5] ). また、これは高価な操作でしょうか?

どのように解決するのですか?

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

同様に

>>> test[1,:]
array([3, 4])

を使用すると、行にアクセスすることができます。 これは、セクション 1.4 (インデックス作成) で説明されている NumPyリファレンス . 少なくとも私の経験では、これは速いです。 ループで各要素にアクセスするよりもずっと速いのは確かです。