複数の配列をcsvファイルに出力したい

Question

実現したいこと

Pythonを使って配列をcsvファイルに出力したいです。

発生している問題・エラーメッセージ

1つの配列は出力することはできたのですが、複数の配列でのやり方がわからず、困っております。

該当のソースコード

import numpy as np
data=np.array([1, 2, 3])
data2=np.array([4, 5, 6])
data3=np.array([7, 8, 9,10])
np.savetxt('transposed.csv', data.T, header='Test', comments='')

補足情報（FW/ツールのバージョンなど）

上に記載したコードでは上のようなcsvファイルが出力されましたが、data2,data3も出力して、下の画像にように2つ以上の配列を入れれるようにしたいです。

1つの配列でのやり方はいくつか見つかったのですが、2つ以上のやり方がわかりません。
ご存知の方いらっしゃいましたら、教えていただきたいです。

Answer 1

numpyは、長さの異なる配列をまとめて扱うのには、あまり向いていません。また、csvの出力についても得意とは言えません。
numpyにこだわりなければ、標準のcsvモジュールか、pandasを使うのが簡単です。
(どちらも itertools.zip_longest を使っています)

import csv
from itertools import zip_longest

with open('transposed.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(['Test', 'Test2', 'Test3'])    # ヘッダー
    writer.writerows(zip_longest(data, data2, data3))

from itertools import zip_longest
import pandas as pd

df = pd.DataFrame(
    zip_longest(data, data2, data3),
    columns=['Test', 'Test2', 'Test3'])
df.to_csv('transposed.csv', index=False)

値が不足している部分はNaNになるので、その列がfloatになります。
それが嫌な場合DataFrameを作るときにdtype=objectを指定してやってください。

Comments

[napo5221]

pandasについて知見が無かったので調べて知識を増やすことができました。ありがとうございました。

Answer 2

import numpy as np

data=np.array([1, 2, 3])
data2=np.array([4, 5, 6])
data3=np.array([7, 8, 9,10])

data_x = [data, data2, data3]
ncol = max(map(len, data_x))
data_x = np.dstack([
    np.pad(i.astype(float), (0, ncol-len(i)), 'constant', constant_values=np.nan)
    for i in data_x
])[0]

data_x = data_x.astype(str)
data_x[data_x == 'nan'] = ''
np.savetxt('transposed.csv', data_x, delimiter=',', fmt='%s', header='Test,Test2,Test3', comments='')

transposed.csv

Test,Test2,Test3
1.0,4.0,7.0
2.0,5.0,8.0
3.0,6.0,9.0
,,10.0

Comments

[napo5221]

stackでまとめて処理するのですね。ありがとうございました。