Pythonと!

プログラミングド素人の勉強日記です。

多次元配列を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')