2 次元配列から特定の列を取得するには?

リスト内包表記またはNumPyライブラリを使用して、2次元配列の特定の列を取得できます。

  1. リスト内包表記を使用する:

もし2次元配列「arr」があった時、i列目を取り出すには次のコードを使用できます

column = [row[i] for row in arr]

なお、i は抜き出す列のインデックスです。

  1. 使用numpy库:

numpy ライブラリがインストール済みであれば、numpy のスライス操作を使用して特定の列を取り出すことができます。2 次元配列 `arr` があり、i 番目の列を取り出したい場合は、以下のコードを使用できます。

import numpy as np
column = arr[:, i]

ここで、i は抽出する列のインデックスです。ただし、numpy のスライス操作では、「:」はすべての要素を表します。

特に大きな配列に対して、NumPyライブラリを使用する方法がより簡潔で効率的です。

bannerAds