python - 在一列上使用多个条件来分配新列的值

Question

我正在尝试根据现有列中的字符串为我的数据分配 8 个标签之一。但是，使用我使用的方法出现此错误:

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

我要查找 144 个不同的字符串，我想将它们分配给 8 个标签。

这是我的意思的一个简化示例。如果 A 是我的数据框中的现有列，我想使用根据 A 的值分配的字符串创建 B。

数据框:

   A     B
0  1   low
1  1   low
2  2   mid
3  3   mid
4  5  high
5  4   mid
6  2   mid
7  5  high

我目前使用的代码是这样的:

for index, row in df.iterrows():
    if df['A'] == 1:
        df['Label'] = 'low'
    elif any([df['A'] == 2, df['A'] == 3, df['A'] == 4]):
        df['Label'] = 'mid'
    elif df['A'] == 5:
        df['Label'] = 'high'

我认为是 any() 的使用给我带来了错误。据我了解，这是因为 pandas 的工作原理，但我不太了解。有没有更简单的方法来做到这一点？

任何帮助或指点将不胜感激:)

Answer 1

这里不需要itterrows，就是bad practice并且被认为很慢。

方法一pd.cut

df['B'] = pd.cut(df['A'], [0,1,4,10], labels=['low', 'mid', 'high'])

   A     B
0  1   low
1  1   low
2  2   mid
3  3   mid
4  5  high
5  4   mid
6  2   mid
7  5  high

---

方法2 np.select

conditions = [
    df['A'] == 1,
    df['A'].isin([2, 3, 4])
]

choices = ['low', 'mid']

df['B'] = np.select(conditions, choices, default='high')

   A     B
0  1   low
1  1   low
2  2   mid
3  3   mid
4  5  high
5  4   mid
6  2   mid
7  5  high