RandomForestRegressor CSVモジュールで予測結果を1列で保存したい

Question

前提・実現したいこと

RandomForestRegressorを用いてコンクリート強度の予測を行っています
そして予測結果をCSVモジュールを使って1列でCSVファイル保存をして、exelファイルに変換したいです。

しかし、予測結果が1行で出力されるので何か方法があれば教えていただきたいです。

該当のソースコード

"予測結果をCSVファイルに保存"
import csv

file = open('y_pred_RF1.csv','w', newline='')
writer = csv.writer(file)
writer.writerow(Y_pred)


file.close()

試したこと

安直な考えでrow（行）をcolumn（列）に変えたりしましたが、案の定出来ませんでした。

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

python3.8 Spyder を使用しています。

###現状の出力

Answer 1

以下のようにreshapeによってn行x1列に変形し、writerowsで書き込めばよいです。
１行１列だけなので、あえてcsvモジュールを使う必要もありませんが。

from sklearn.ensemble import RandomForestRegressor
import numpy as np

# テストデータの作成
# 決定木とランダムフォレスト
# https://qiita.com/yshi12/items/6d30010b353b084b3749

def sin_model(x, sigma=0.2):
    """大きな波＋小さな波＋ノイズからなるダミーデータ。"""
    noise = sigma * np.random.randn(len(x))
    return np.sin(5 * x) + np.sin(0.5 * x) + noise

x = 10 * np.random.rand(100)
y = sin_model(x)
rfr = RandomForestRegressor(100)
rfr.fit(x[:, None], y)
xfit = np.linspace(0, 10, 10)
Y_pred = rfr.predict(xfit[:, None])


"予測結果をCSVファイルに保存"
import csv

Y_pred = Y_pred.reshape(-1,1) # n行x1列に変形

file = open('y_pred_RF1.csv','w', newline='')
writer = csv.writer(file)
writer.writerows(Y_pred) # ～row -> ～rows で複数行書込
file.close()

Comments

[xtail]

解決しました!ありがとうございます