Python / Pandas 条件新列 ValueError

Question

我希望向现有数据帧添加一个新列，并根据df.Public(1|2( 的值将df.TaxStatus设置为 True 或 False。我看到其他人使用Numpy.where()创建新列，但我收到 ValueError。

df['TaxStatus'] = np.where(df.Public == 1, True, np.where(df.Public == 2, False))

这似乎有效，除了 ValueError：either both or neither of x and y should be given

Answer 1

你可以简单地说：

df['TaxStatus'] = df.Public == 1

表达式df.Public == 1返回一系列 True/False 值，您可以将这些值分配给新列。无需np.where()