pandas aggregate count в dataframe

Question

У меня есть DataFrame, и я использую .aggregate({'col1': np.sum}), это будет выполнять суммирование значений в col1 и объединить их вместе. Можно ли выполнить подсчет, что-то вроде .aggregate({'col1': some count function here})?

Answer 1

Вы можете использовать 'size', 'count', или 'nunique' в зависимости от вашего прецедента. Различия между ними являются:

• 'size': отсчет включая NaN и повторных значений.
• 'count': кол-во, исключая NaN, но включая повторы.
• 'nunique': количество уникальных значений, исключая повторы и NaN.

В качестве примера рассмотрим следующую DataFrame:

df = pd.DataFrame({'col0': list('aabbcc'), 'col1': [1, 1, 2, np.nan, 3, 4]}) 

    col0 col1 
0 a 1.0 
1 a 1.0 
2 b 2.0 
3 b NaN 
4 c 3.0 
5 c 4.0 

Затем, используя три функции, описанные в:

df.groupby('col0')['col1'].agg(['size', 'count', 'nunique']) 

     size count nunique 
col0      
a  2  2  1 
b  2  1  1 
c  2  2  2