將結果導出到 csv 並在一列中打印

Question

我有以下代碼，我預測第二天是否會下雨。 結果我得到1或0。

weatherbayes = GaussianNB()
weatherbayes.fit(X_train, y_train)
predbayes = weatherbayes.predict(df_test)
data = [predbayes]
print(data)
np.savetxt("student.csv",data, newline =" ", delimiter = '\t',  fmt ='% s')

我得到以下結果：

數據形狀：(1, 11540)

但我需要將它打印在一列中，並且為每個值打印一行。

Answer 1

如果您的形狀是 (1, 11540) 並且您希望每行一個值，請轉置您的數組。

# data = np.random.choice([0, 1], (1, 10))
# data.shape -> (1, 10)
np.savetxt('student.csv', data.T, fmt='%d')

student.csv文件內容：

0
1
0
1
0
1
1
1
1
0