【Python】2次元配列から指定した行だけ抽出する方法

eycatch

叙々苑のドレッシングがブームのmizkyです。

今回は、Pythonにて2次元配列から指定した行だけ抽出(取り出す)方法について書きたいと思います。

具体例を挙げると

配列[[1 ,2, 3],[100, 200, 300],[10000, 20000, 30000]]から

1列目を[1, 100, 10000]のように取り出す方法の紹介となります。

私が失敗した方法

array = [[1,2,3],[100,200,300],[10000,20000,30000]]

column0 = array[:][0]
print(column0)

このソースコードだと普通に

[1, 2, 3]

と1行目が取り出されてしまいます。

指定した行を取り出す方法

1行目を取り出す場合は、以下のように記述すれば取り出せます。

array = [[1,2,3],[100,200,300],[10000,20000,30000]]

column0 = [row[0] for row in array]
print(true_column0)

出力結果は

[1, 100, 10000]

となります。

2次元配列の各行から1個目の要素を選択して、新しい配列に格納しているといったソースコードとなります。

所感

久々にPythonを触りましたが、やはり直感で書けるので分かりやすいですね。

コメント

タイトルとURLをコピーしました