列表理解使用多列

Question

我有一个pandas数据框，其中包含实际的列，并进行了预测。我想使用列表理解来创建一个新列，当实际值=预测时= 1，否则为0。我知道如何使用np.where做到这一点，但我很想知道如何使用列表理解来做到这一点。

这可以使用np.where：

合并[＆＃39;正确＆＃39;] = np.where（combined.actual == combined.predicted，1,0）

谢谢！

Answer 1

你不需要np.where或list comprehension：

您可以使用：

combined['correct'] = (combined.actual == combined.predict).mul(1)

或

combined['correct'] = (combined.actual == combined.predict).astype(int)