Wie zählen Sie die eindeutigen Werte jeder Zeile in einer Spalte mit Python?

Question

Ich habe einen solchen Datenrahmen:

id countries 
01 [UK,UK,UK,US] 
02 [US,US,US,US] 
03 [FR,UK,CN,US] 

Ich möchte, wie viele Länder für jede ID existieren zählen. Wie sollte das Ergebnis wie:

id countries counts 
01 [UK,UK,UK,US] 2 
02 [US,US,US,US] 1 
03 [FR,UK,CN,US] 4 

Answer 1

Wenn Werte sind list s sie set konvertieren und erhalten length:

print (type(df.loc[0, 'countries'])) 
<class 'list'> 

df['counts'] = df['countries'].apply(lambda x: len(set(x))) 
print (df) 
    id   countries counts 
0 1 [UK, UK, UK, US]  2 
1 2 [US, US, US, US]  1 
2 3 [FR, UK, CN, US]  4 

Oder wenn Werte strings sind erste [] und Spaltung entfernen:

print (type(df.loc[0, 'countries'])) 
<class 'str'> 

df['counts'] = df['countries'].str.strip('[]').str.split(',').apply(lambda x: len(set(x))) 
print (df) 
    id  countries counts 
0 1 [UK,UK,UK,US]  2 
1 2 [US,US,US,US]  1 
2 3 [FR,UK,CN,US]  4