我有一个pandas数据框，其中包含实际值列和预测值。我想使用列表推导创建一个新列，当实际值=预测时= 1，否则返回0。我知道如何使用np.where来做到这一点，但是我很好奇知道如何使用列表理解来做到这一点。
使用np.where可以工作：
combined['correct'] = np.where(combined.actual==combined.predicted, 1, 0)
谢谢！
最佳答案
您不需要np.where也不需要列表理解：
您可以使用此：
combined['correct'] = (combined.actual == combined.predict).mul(1)
要么
combined['correct'] = (combined.actual == combined.predict).astype(int)