我的目标是列出我的:
predictions = [0, 0.2, 0.9, 0.7]
如果高于0.5,则应为0,如果不是1
我试过:

predictions = np.where(predictions>=0.5,1, 0).tolist()

但当采取第一项措施时:
[0]
不仅如此
做我想做的事最好的方法是什么?

最佳答案

您可以使用列表理解:

[0 if i>=0.5 else 1 for i in predictions]

注意:您说过如果原始值高于0.5,则条目为零。你确定吗?如果低于0.5,应该是零吗?

10-07 15:08