Python Pandas：AttributeError：'str'对象没有属性'loc'

Question

使用数据框 df：

Count
1
2
3
4
5

想要添加第二列，将 3 以上的所有内容归类为“4+” - 需要的输出：

Count | Category
1        1
2        2
3        3
4        4+
5        4+

这是我的代码：

df['Category'] = df['Count']
df = df.loc[df['Count'] > 3, 'Category'] = '4+'

我得到这个错误：

AttributeError: 'str' object has no attribute 'loc'

Answer 1

随走

df['Category'] = df['Count']
df.loc[df['Count'] > 3, 'Category'] = '4+'