[python]配列の出力について

Question

#配列のcsv出力について

list1 = [1,2,3,"",45]
list2 = [4,"",6,89,""]

data = [ list1 + list2]

上記の「data」の配列をcsvに出力する際に下記の様に縦に出力したいです。

どのような方法を使えば、どのような調べ方をすればいいかもわからずご質問させて頂きました。
お手数をおかけしますがよろしくお願いいたします。

Answer 1

csv モジュールの csv.writer() を使って行ごとに書き込めばいいです。

import csv

list1 = [1,2,3,"",45]
list2 = [4,"",6,89,""]

with open('output.csv', 'w') as f:
    writer = csv.writer(f)
    for row in zip(list1, list2):
        writer.writerow(row)

Comments

[Dadada_c.]

ありがとうございます！なるほど…自分でリストをまとめてしまってややこしくしてました；；
ありがとうございました！

Answer 2

[ list1 + list2]
でなく
[list1, list2]
として、２次元配列 (行列) にしてやれば、
[x][y] の位置の要素を [y][x] にしてやる (行列の転置操作)ことで実現できます。

また、 writerows の一行で ２次元配列をすべて出力できます。

mat.py

import csv

list1 = [1,2,3,"",45]
list2 = [4,"",6,89,""]

mat = [list1, list2]
mat_t = [[mat[y][x] for y in range(2)] for x in range(5)]

out_path = "out.csv"

with open(out_path, 'w') as f:
    writer = csv.writer(f)
    writer.writerows(mat_t)

実行例

zip 関数をつかったり、 numpy で行列操作しても同じ結果を得ることができます。

参考情報

• Pythonリスト型の二次元配列の行と列を入れ替える（転置）

https://note.nkmk.me/python-list-transpose/

• Pythonでcsvから行方向のデータを読み込みたい

https://teratail.com/questions/138897

Comments

[Dadada_c.]

なるほど！こんなありかたもあるんですね。とても勉強になりました！ありがとうございました！