为什么将 df 导出到 cvs 不导出第二列？

Question

我有一个包含索引的 df 和一个浮点数并导出到 csv 的列。 但是，当我从浮点数更改为整数，然后将列的值更改为 1（对于那些值高于 50 的值和 0 的值低于 50 的值）然后导出到 csv 失败并且仅导出索引 plust 列标题。 我已经调试并得出结论，问题出在我输入“0”值时。到那时 csv 导出工作.... 有什么提示吗？

谢谢

prediction = model.predict(predict_data)
sol = pd.DataFrame(data=prediction, columns=['predicted'])
sol.index.name = 'index'
sol['final_prediction'] = final_sol['predicted'].astype(int)
sol.loc[sol['predicted'] > 50, 'predicted'] = 1
sol.loc[sol['predicted'] == 50, 'predicted'] = 1
sol.loc[sol['predicted'] < 50, 'predicted'] = 0  # this is the point when fail occurs. 
sol.to_csv('export_file.csv')

Answer 1

sol = pd.DataFrame(data=task, columns=['predicted'])
sol.index.name = 'index'
sol['final_prediction'] = sol['predicted'].astype(int)
sol.loc[sol['predicted'] > 50, 'predicted'] = 1
sol.loc[sol['predicted'] == 50, 'predicted'] = 1
sol.loc[sol['predicted'] < 50, 'predicted'] = 0  # this is the point when fail occurs. 
# print(sol)
sol.to_csv('export_file.csv')

我认为这一行有错误final_sol['predicted'].astype(int)