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

问题描述

我有一个包含索引的df和一个浮点数的列，并导出到csv作品。 但是，当我从浮点数更改为整数，然后将值大于50的列的值更改为1，将值小于50的列的值更改为0时，则导出到csv失败，并且仅导出索引加号列标题。 我已经调试并得出结论，问题出在我输入“ 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')

解决方法

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)上的错误