2 次元配列から特定の列を取得するには?
リスト内包表記またはNumPyライブラリを使用して、2次元配列の特定の列を取得できます。
- リスト内包表記を使用する:
もし2次元配列「arr」があった時、i列目を取り出すには次のコードを使用できます
column = [row[i] for row in arr]
なお、i は抜き出す列のインデックスです。
- 使用numpy库:
numpy ライブラリがインストール済みであれば、numpy のスライス操作を使用して特定の列を取り出すことができます。2 次元配列 `arr` があり、i 番目の列を取り出したい場合は、以下のコードを使用できます。
import numpy as np
column = arr[:, i]
ここで、i は抽出する列のインデックスです。ただし、numpy のスライス操作では、「:」はすべての要素を表します。
特に大きな配列に対して、NumPyライブラリを使用する方法がより簡潔で効率的です。