NumPy の多次元配列から特定の列を抽出するにはどうすればよいですか?

NumPy で多次元配列から列を取得する

NumPy の基本的な操作は、多次元配列内の個々の要素またはデータのサブセットにアクセスすることです。構文 test[i] を使用して行を取得するのは簡単ですが、特定の列の抽出は少し異なる場合があります。

NumPy 配列の i 番目の列にアクセスするには、構文 test[:, i] を利用できます。これは、元の行列の i 番目の列の値を含む 1 次元配列を返します。

たとえば、次の配列を考えてみましょう。

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

最初の列 (を含む) を取得するには、値 1、3、および 5) を取得するには、次を使用します:

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

同様に、2 番目の列 (値 2、4、および 6 を含む) を取得するには、次を使用します:

>>> test[:, 1]
array([2, 4, 6])

この操作は特に計算コストがかかるわけではなく、NumPy で多次元データを操作するための一般的な手法であることに注意してください。

以上がNumPy の多次元配列から特定の列を抽出するにはどうすればよいですか?の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。