Pythonで2次元配列の中のn番目の列だけを取り出し、その要素を3個ずつ別のリストに格納する方法

Question

前提

Pythonを用いて、2次元配列の中の特定の列だけを取り出し、その要素を3個ずつ別の2次元配列に格納したいのですが、上手くいきません...。

a = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], 
     [11, 12], [13, 14], [15, 16], [17, 18]]

というような形式のデータの中の1番目の列のみを取り出し、

b = [[1, 3, 5], [7, 9, 11], [13, 15, 17]]

このように要素を3個ずつ格納した新たな2次元配列を作成したいのですが、どのようにプログラムを組めばよいでしょうか。

初学者ゆえ初歩的な質問ではございますが、教えていただけますと幸いです。宜しくお願いいたします。

実現したいこと

▲▲機能を動作するようにする

試したこと

空のリストを作成し、for文を用いて二重ループで値を追加する方法や、内包表記を使った書き方があるというのは理解したのですが、試していくうちに混乱してしまい、なかなか正解に辿り着けませんでした...。

print([i[0] for i in a])

上記で1番目の列のみの抽出までできましたが、3個ずつの格納はできませんでした。

お手数をおかけしますが、記述方法をご教授いただければ幸いです。

Answer 1

ライブラリとか使わない、泥くさいやりかた。

a = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], 
     [11, 12], [13, 14], [15, 16], [17, 18]]

# すべてのリストの先頭だけのリスト
firsts = [x[0] for x in a]

# リストを3つずつのリストのリストにする。
first_partitioned = [firsts[i:i+3] for i in range(0, len(firsts), 3)]

Answer 2

numpyを利用して

import numpy as np

x = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12], [13, 14], [15, 16], [17, 18]])
print(x.T.reshape(-1, 3))

Answer 3

more_itertools の chunked を使う場合。

from more_itertools import chunked

a = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], 
     [11, 12], [13, 14], [15, 16], [17, 18]]
b = [*chunked([i[0] for i in a], 3)]

print(b)

Comments

[masora]

迅速にご教示いただきありがとうございました。
無事にコードを組むことが出来ました。