为每个国家计算一定的价值

debugcn 发表于 Dev

杰克·瓦格纳

我正在尝试执行Excelcountif函数，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吐出一个总值，而不是显示的5所有值Quarter_1。我忘了如何通过下面的方式计算我的函数Country？请对此提供任何帮助！

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

二凡

我们可以使用您的专栏的长度，并进行地板分割来创建您的宿舍。然后，我们将这些分组并取总和。

最后，我们添加前缀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

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。