Pandas: для каждого элемента в столбце подсчитайте количество разных элементов в другом столбце

Question

В данном фреймворке данных Pandas мне нужно подсчитать количество уникальных элементов в столбце y для каждого элемента в столбце x. Я придумал методы подсчета в dicts, но они, похоже, не работают должным образом. Пример того, что я пытаюсь управлять:

df = pd.DataFrame([[a, 2], [a, 1], [c, 4], [a, 1], [c, 4]]), columns=list('AB')) 

Который создает следующий Dataframe:

A B 
0 a 2 
1 a 1 
2 c 4 
3 a 1 
4 c 4 

Теперь выход я хочу должен быть:

a:2 
c:1 

Answer 1

Вам нужно groupby с SeriesGroupBy.nunique и последний to_dict:

print (df.groupby('A').B.nunique()) 
A 
a 2 
c 1 
Name: B, dtype: int64 

print (df.groupby('A').B.nunique().to_dict()) 
{'a': 2, 'c': 1} 

Answer 2

Используйте drop_duplicates, чтобы найти уникальные значения. Затем используйте value_counts

df.drop_duplicates().A.value_counts() 

a 2 
c 1 
Name: A, dtype: int64