Groupby 并在 Pandas 中打印整个数据框答案

【问题标题】：Groupby and print entire dataframe in PandasGroupby 并在 Pandas 中打印整个数据框
【发布时间】：2021-12-17 16:36:09
【问题描述】：

我有一个如下的数据集，在这种情况下，我想统计每个国家的水果数量，并输出为数据集中的一列。

我尝试使用 groupby， df=df.groupby('Country')['Fruits'].count(),

但在这种情况下，我没有得到预期的结果，因为 groupby 只输出计数而不是整个数据帧/数据集。

如果有人能提出更好的方法来做这件事会很有帮助。

数据集

Country     Fruits      Price       Sold            Weather     

India       Mango       200         Market          sunny
India       Apple       250         Shops           sunny
India       Banana      50          Market          winter
India       Grapes      150         Road            sunny
Germany     Apple       350         Supermarket     Autumn
Germany     Mango       500         Supermarket     Rainy
Germany     Kiwi        200         Online          Spring
Japan       Kaki        300         Online          sunny
Japan       melon       200         Supermarket     sunny

预期输出


Country     Fruits      Price       Sold            Weather     Number

India       Mango       200         Market          sunny       4
India       Apple       250         Shops           sunny       4
India       Banana      50          Market          winter      4
India       Grapes      150         Road            sunny       4
Germany     Apple       350         Supermarket     Autumn      3
Germany     Mango       500         Supermarket     Rainy       3
Germany     Kiwi        200         Online          Spring      3
Japan       Kaki        300         Online          sunny       2
Japan       melon       200         Supermarket     sunny       2

谢谢你:)

【问题讨论】：

标签： python-3.x pandas dataframe pandas-groupby

【解决方案1】：

您正在寻找transform:

df['count'] = df.groupby('Country')['Fruits'].transform('size')

   Country  Fruits  Price         Sold Weather  count
0    India   Mango    200       Market   sunny      4
1    India   Apple    250        Shops   sunny      4
2    India  Banana     50       Market  winter      4
3    India  Grapes    150         Road   sunny      4
4  Germany   Apple    350  Supermarket  Autumn      3
5  Germany   Mango    500  Supermarket   Rainy      3
6  Germany    Kiwi    200       Online  Spring      3
7    Japan    Kaki    300       Online   sunny      2
8    Japan   melon    200  Supermarket   sunny      2

【讨论】：