熊猫分组和计数答案

【问题标题】：Pandas Group By and Count熊猫分组和计数
【发布时间】：2017-03-21 17:10:02
【问题描述】：

熊猫数据框 df 有 3 列：

user_id, 会议，收入

我现在要做的是按唯一 user_id 对 df 进行分组并派生 2 个新列 - 一个称为 number_sessions（计算与特定 user_id 关联的会话数），另一个称为 number_transactions（计算收入列下的行数每个 user_id 的值 > 0）。我该怎么做？

我试着做这样的事情：

df.groupby('user_id')['session', 'revenue'].agg({'number sessions': lambda x: len(x.session), 
'number_transactions': lambda x: len(x[x.revenue>0])})

【问题讨论】：

没有问题或问题陈述

标签： python pandas count group-by apply

【解决方案1】：

我认为你可以使用：

df = pd.DataFrame({'user_id':['a','a','s','s','s'],
                   'session':[4,5,4,5,5],
                   'revenue':[-1,0,1,2,1]})

print (df)
   revenue  session user_id
0       -1        4       a
1        0        5       a
2        1        4       s
3        2        5       s
4        1        5       s

a = df.groupby('user_id') \
      .agg({'session': len, 'revenue': lambda x: len(x[x>0])}) \
      .rename(columns={'session':'number sessions','revenue':'number_transactions'})
print (a)
         number sessions  number_transactions
user_id                                      
a                      2                    0
s                      3                    3

a = df.groupby('user_id') \
      .agg({'session':{'number sessions': len}, 
            'revenue':{'number_transactions': lambda x: len(x[x>0])}}) 
a.columns = a.columns.droplevel()

print (a)
         number sessions  number_transactions
user_id                                      
a                      2                    0
s                      3                    3

【讨论】：

非常感谢！这正是我需要的解决方案。干杯！

【解决方案2】：

我会使用 nunique 代替 session，以免重复计算特定用户的同一会话

funcs = dict(session={'number sesssions': 'nunique'},
             revenue={'number transactions': lambda x: x.gt(0).sum()})
df.groupby('user_id').agg(funcs)

设置

df = pd.DataFrame({'user_id':['a','a','s','s','s'],
                   'session':[4,5,4,5,5],
                   'revenue':[-1,0,1,2,1]})

【讨论】：