csv書き込みでdtypeエラーが発生する

Question

前提・実現したいこと

学習した結果をcsvに書き込んで保存したいのですが、書き込む過程でエラーが出てきて保存できません。

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

Traceback (most recent call last):
  File "/home/deepstation/PycharmProjects/hoge/hogehoge.py", line 444, in <module>
    writer.writerow([i] + row)
TypeError: ufunc 'add' did not contain a loop with signature matching types dtype('<U32') dtype('<U32') dtype('<U32')

該当のソースコード

P_list = []

#学習・予測部分省略

    pred_P = model.predict(P_valid)
    for i in range(10):
        P_rmse = np.sqrt(np.mean(pred_P[i] - P_valid[i])**2)
        P_list.append(P_rmse)

index = ['1', '2', '3', '4', '5', '6', '7', '8']
with open(CSV_fname + '.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerow(header)
    for i, row in zip(index, P_list):
        writer.writerow([i] + row)

試したこと

デバッグモードで確認したところ、P_list、indexともにクラスはリストです。pred_Pはndarrayです。

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

Answer 1

writer.writerow([i] + row)

の部分では ndarray である row の各要素に i を加算する処理が呼び出されてしまいます。
i は文字列になっているのでエラーが発生しています。
リスト同士の + であれば連結になるので次のように tolist() を使えばよいと思います。

writer.writerow([i] + row.tolist())

Comments

[退会済みユーザー]

ありがとうございます。このように実行したところ
writer.writerow([i] + row.tolist())
TypeError: can only concatenate list (not "float") to list
というエラーが出現してしまいました。

[kit494way]

すみません。row が ndarray だと早合点しました。
row は float で、1行に2列（行番号と数値）の csv を出力しようとしている、という理解で良いでしょうか？
であれば、
writer.writerow([i, row])
でよいかと思います。

[退会済みユーザー]

説明が足りずに申し訳ございません。P_listは8行10列の行列です。そのため、writer.writerow([i, row])を試したところ1行につき2列の情報が入力されていましたが、出力は1行につき11列（行番号と数値）の情報を記載したいです。お手数をお掛けして申し訳ございません。

[kit494way]

P_list が行列ではなく、1次元リストになっていないか、 print(P_list) で確認してみてください。

[退会済みユーザー]

確認したところ、1次元のリストになっていました。整形したところうまく出来ました。ありがとうございました！