计算每个国家的某个值

Question

我正在尝试用 pandas 做一个 Excel countif pandas但这样做遇到了障碍。

我有这个dataframe 。 我需要按季度计算每个国家/地区的YES 。 我已经在下面发布了要求的答案。

result.head(3)

Country    Jan 1    Feb 1   Mar 1   Apr 1   May 1   Jun 1       Quarter_1         Quarter_2 
FRANCE     Yes      Yes      No      No     No       No            2                  0
BELGIUM    Yes      Yes      No     Yes     No       No            2                  1
CANADA     Yes      No       No     Yes     No       No            1                  1

我尝试了以下方法，但Pandas吐出一个总值，而不是为Quarter_1下的所有值显示5 。 我不知道如何按Country计算我的 function？ 请对此提供任何帮助！

result['Quarter_1'] = len(result[result['Jan 1'] == 'Yes']) + len(result[result['Feb 1'] == 'Yes'])
+ len(result[result['Mar 1'] == 'Yes'])

Answer 1

我们可以使用您的柱子的长度并按楼层划分来创建您的宿舍。 然后我们对这些进行分组并求和。

最后我们添加前缀Quarter ：

df = df.set_index('Country')
grps = np.arange(len(df.columns)) // 3
dfn = (
    df.join(df.eq('Yes')
            .groupby(grps, axis=1)
            .sum()
            .astype(int)
            .add_prefix('Quarter_'))
    .reset_index()
)

或使用列表推导重命名您的列：

df = df.set_index('Country')
grps = np.arange(len(df.columns)) // 3

dfn = df.eq('Yes').groupby(grps, axis=1).sum().astype(int)
dfn.columns = [f'Quarter_{col+1}' for col in dfn.columns]

df = df.join(dfn).reset_index()

   Country Jan 1 Feb 1 Mar 1 Apr 1 May 1 Jun 1  Quarter_1  Quarter_2
0   FRANCE   Yes   Yes    No    No    No    No          2          0
1  BELGIUM   Yes   Yes    No   Yes    No    No          2          1
2   CANADA   Yes    No    No   Yes    No    No          1          1