python - 在函数中使用 groupby 组名

Question

我有这样的数据:

df = pd.DataFrame({'user': np.random.choice(['a', 'b','c'], size=100, replace=True),
                   'value1': np.random.randint(10, size=100),
                   'value2': np.random.randint(20, size=100)})

我用它来产生一些结果，例如，

grouped = df.groupby('user')
results = pd.DataFrame()
results['value2_sum'] = grouped['value2'].sum()

对于此结果数据框的其中一列，我想将用户名传递给另一个函数，该函数使用数据框外部的数据。

我试过类似的方法:

results['user_result'] = grouped.apply(lambda x: my_func(x.index))

但找不到有效的语法。

Answer 1

您希望 .name 属性访问组索引值:

In [6]:
grouped = df.groupby('user')
results = pd.DataFrame()
results['value2_sum'] = grouped['value2'].sum()
results['user_result'] = grouped.apply(lambda x: x.name)
results

Out[6]:
      value2_sum user_result
user                        
a            342           a
b            333           b
c            308           c