如何在python中的逻辑回归模型中将预测值保存为字符串

Question

这段代码给我的预测值是1 0r 0。但我想把它们打印成"成功"或"失败"。有办法吗？

prediction = pd.DataFrame(predictions2, columns=['predictions']).to_csv('prediction.csv')

Answer 1

您所需要做的就是使用字典将0和1映射到Failure和Success

prediction = pd.DataFrame(predictions2, columns=['predictions'])
prediction['predictions'] = prediction['predictions'].astype(int)
prediction["predictions"] = prediction.predictions.map({0:'Failure', 1:'Success'})
prediction.to_csv('prediction.csv', index=False)

你必须确保"预测"的数据类型是内部的。

Answer 2

我对panda没有太多经验，但如果你只想将这些值分配给这些输出，一种方法是使用字典(按我键入的方式发布(。您也可以使用三元运算符。例如：

prediction = pd.DataFrame(predictions2, columns ['predictions']).to_csv('prediction.csv')
result = 'Success' if prediction == 1 else 'Failure'
print(result)

这可以进一步简化为：

print('Success' if prediction == 1 else 'Failure')

注意：如果输出是int，请使用1；如果是字符串，请使用"1"。

Answer 3

谢谢大家。问题解决了！

**mydict ={0:'Failure',1:'Success'}
predictions2 = [mydict[i] for i in predictions2]
print(predictions2)
predictions2= pd.DataFrame(predictions2, columns=['predictions']).to_csv('prediction3.csv', index=False)