多次元配列をpandasのDataFrameにする方法

はじめに

他クラス分類の結果を提出形式にするための自分用メモ。

方法

y_pred: n 行 ×m 列の numpy array

# データフレームに変換
submission = pd.DataFrame(y_pred)

# 一番左の列に列名'id'でtest\_dfの'id'列を追加する。
submission.insert(0,'id' , test_df['id'])

# submission.csvに出力。ヘッダーなし。インデックスなし。
submission.to_csv('submission.csv', header=False, index=False, encoding='utf-8')