python - 使用多列的列表理解

Question

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

这可以使用 np.where:

combined['correct'] = np.where(combined.actual==combined.predicted, 1, 0)

谢谢!

Answer 1

你不需要 np.where 也不需要列表理解:

你可以使用这个:

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

或

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