根据条件在熊猫数据帧中计算值发生的次数。

Question

我试图计算一个值在数据帧中的特定位置上发生了多少次。

举个例子，我使用了这个数据框架：

   import pandas as pd
   
   d = {'Fruit': ['Apple', 'Apple', 'Apple', 'Onion', 'Onion', 'Onion', 'Onion', 'Pear', 'Pear', 'Pear', 
   'Pear', 'Pear'],
        'Country': ['USA', 'SUI', 'USA', 'SUI', 'USA', 'SUI', 'SUI', 'USA', 'USA', 'USA', 'SUI', 'SUI']}

   df = pd.DataFrame(data=d)

我不明白如何计算，例如，在美国和SUI中有多少个苹果，并将其添加到“计数”列中。

输出应该如下所示：

import pandas as pd

d = {'Fruit': ['Apple', 'Apple', 'Apple', 'Onion', 'Onion', 'Onion', 'Onion', 'Pear', 'Pear', 'Pear', 'Pear', 'Pear'],
     'Country': ['USA', 'SUI', 'USA', 'SUI', 'USA', 'SUI', 'SUI', 'USA', 'USA', 'USA', 'SUI', 'SUI'],
     'Count': [2, 1, 2, 3, 1, 3, 3, 3, 3, 3, 2, 2]}

df = pd.DataFrame(data=d)

我知道如何计算值本身(果树列中有多少苹果)，但不知道如何将此条件添加到计算中。

谢谢你提前帮忙。

Answer 1

试试群比变换

df['counts'] = df.groupby(['Fruit', 'Country'])['Country'].transform('size')

df

    Fruit Country  counts
0   Apple     USA       2
1   Apple     SUI       1
2   Apple     USA       2
3   Onion     SUI       3
4   Onion     USA       1
5   Onion     SUI       3
6   Onion     SUI       3
7    Pear     USA       3
8    Pear     USA       3
9    Pear     USA       3
10   Pear     SUI       2
11   Pear     SUI       2

Answer 2

您可以使用groupby后跟一个join，如下所示：

fruit_counts = df.groupby(["Fruit", "Country"]).size().rename("Count")
df.join(fruit_counts, on=["Fruit", "Country"])

输出：

    Fruit Country  Count
0   Apple     USA      2
1   Apple     SUI      1
2   Apple     USA      2
3   Onion     SUI      3
4   Onion     USA      1
5   Onion     SUI      3
6   Onion     SUI      3
7    Pear     USA      3
8    Pear     USA      3
9    Pear     USA      3
10   Pear     SUI      2
11   Pear     SUI      2