Python Pandas: Attribute: 'str' Objekt hat kein Attribut 'loc'

Question

Arbeiten mit Datenrahmen df:

Count 
1 
2 
3 
4 
5 

Wollen zweite Spalte hinzuzufügen, stuft, dass alles über 3 als '4+' - benötigt Ausgang:

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

Dies ist mein Code:

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

Und ich bekomme diese Fehlermeldung:

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

Answer 1

Gehen Sie einfach mit

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

Answer 2

Sie können versuchen, mit:

import pandas as pd 
df = pd.DataFrame({"Count": [1,2,3,4,5]}) 
df["Category"] = df["Count"].apply(str) 
df["Category"][df['Count'] > 3] = "4+" 

Ausgabe wäre:

>>> df 
    Count Category 
0  1  1 
1  2  2 
2  3  3 
3  4  4+ 
4  5  4+