Как подсчитать количество элементов группы с помощью pandas

Question

У меня есть dataframe и просто хочу подсчитать количество элементов в каждой группе. Я знаю, я могу использовать groupby(). Count(), чтобы получить все подсчеты всех столбцов, но для меня это слишком много, мне просто нужно количество элементов в каждой группе. Как я могу это сделать?

Вот пример:.

mydf = pd.DataFrame({"fruit":["apple","banana","apple"],"weight":[7,8,3],"price":[4,5,6]}) 
mydf 
>>  fruit price weight 
>> 0 apple  4  7 
>> 1 banana  5  8 
>> 2 apple  6  3 

Если я использую GroupBy ("фрукты") означает,(), я получаю значение для каждого столбца.

mydf.groupby("fruit").mean() 

>>   price weight 
>> fruit     
>> apple  2  2 
>> banana  1  1 

Но мой ожидать выход:

>>   number_of_fruit 
>> fruit     
>> apple  2 
>> banana  1 

Как я могу это сделать?

Answer 1

Вы хотите size подсчитать количество каждого плода:

In [102]: 
mydf.groupby('fruit').size() 

Out[102]: 
fruit 
apple  2 
banana 1 
dtype: int64