根据熊猫中的第三列在两列之间进行选择

Question

我有一个 Pandas 数据框，其中包含三列 A、B、C。我想将 D 列添加到数据框中，其值根据 C 中的值取自 A 和 B。例如，如果 C

Answer 1

您可以为此使用np.where：

import numpy as np

df['D'] = np.where(df['C'].lt(0.5),df['B'],df['A'])

这读作，对于每一行，如果 C 中的值小于 0.5，则从 B 列返回值，否则从 A 列返回值。

请注意，您可以将lt(0.5) 替换为le(0.5) 以表示小于或等于0.5。