根据具有三个选项的其他列值更改熊猫数据框列值

Question

我有一个 Pandas 数据框，其中有一列名为 col1，我想在数据框中添加一个新列，该列的值取决于每条记录的 col1 的内容。我试过这个：

d['new_column'] = 'CASE 1' if d['COL1'] == None else 'CASE 2' if d['COL1'] == 0 else 'CASE 3'

但是我收到了这个错误：

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

如何进行评估以正确设置新列的值？

Answer 1

尝试使用np.where。在这种情况下，它既高效又易于使用：

import numpy as np

df['new_column'] = np.where(d['COL1'].isnull(), 'CASE1',
                            np.where(d['COL1']==0, 'CASE 2', 'CASE 3'))