pythonで複数のリストの各要素を取り出して二次元配列を作りたい

Question

pythonで複数のリストの各要素を取り出して二次元配列を作りたい

pythonに関する質問です。
time = [1,2,3,....,30]
freq = [10,20,30,...,300]
Pxx = [0.1,0.2,...,3.0]
のように各長さが同じなlistが3つあったとき，

これをtime, freq, Pxxのそれぞれ各要素を1番目から順に取っていき
data = [[1,10,0.1],[2,20,0.2],...]
という二次元配列を新たに作成したいです。

C言語だと
for(i = 0, i<30 , i++){
fprintf(fp,"%f %f %f\n"time[i], freq[i], Pxx[i]);
}

のようにファイルに書き込んでいけばいいのですが、listではどのように行えばよろしいでしょうか。

Answer 1

zipを使うと簡単です。

time = [1,2,3,30]
freq = [10,20,30,300]
Pxx = [0.1,0.2,3.0]

data = [[t, f, p] for t, f, p in zip(time,freq,Pxx)]
print(data)
# [[1, 10, 0.1], [2, 20, 0.2], [3, 30, 3.0]]

追記

n番目まで（ただし計算は最後までされてしまいます）

time = [1,2,3,30]
freq = [10,20,30,300]
Pxx = [0.1,0.2,3.0]
n = 1
data = [[t, f, p] for i, (t, f, p) in enumerate(zip(time,freq,Pxx)) if i <= n]
print(data)
# [[1, 10, 0.1], [2, 20, 0.2]]

元listの要素が多すぎる場合はこちら

time = [1,2,3,30]
freq = [10,20,30,300]
Pxx = [0.1,0.2,3.0]
n = 1
data = list()
for i, (t, f, p) in enumerate(zip(time,freq,Pxx)):
    if i > n:
        break
    data.append([t, f, p])
print(data)
# [[1, 10, 0.1], [2, 20, 0.2]]

Comments

[Lily1007]

zipを使うと上手くいきました。ありがとうございます！
また、zipをもちいてtime, freq, Pxxのそれぞれn番目までの要素を取るという操作はそもそも可能なのでしょうか。調べて見ても、特定の要素数を引数としてzipを用いる例が無かったもので…

[jeanbiego]

追記しました。
計算量に応じて好きな方を使ってください。

[Lily1007]

追記に関しても確認しました。誠にありがとうございます。

[actorbug]

同様の処理はitertoolsモジュールのisliceを使っても可能です。
import itertools
data = [list(t) for t in itertools.islice(zip(time,freq,Pxx), n)]