将结果导出到 csv 并在一列中打印

Export result to csv and print it in one column

我有以下代码,可以预测第二天是否会下雨。结果我得到 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)

但我需要将其打印在一列中,每个值打印一行。

如果您的形状是 (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