python - 在 pandas 中使用 .groupby() 时如何输出 groupby 变量？

Question

我有一些数据想要分析。我按相关组变量(此处为“test_condition”和“region”)对数据进行分组，并使用我编写的函数分析测量变量(“rt”):

grouped = data.groupby(['test_condition', 'region'])['rt'].apply(summarize)

效果很好。输出如下所示(假数据):

                                           ci1         ci2        mean  
test_condition      region                                               
Test Condition Name And          0  295.055978  338.857066  316.956522   
                    Spill1       0  296.210167  357.036210  326.623188   
                    Spill2       0  292.955327  329.435977  311.195652   

问题是，“test_condition”和“region”不是实际的列，我无法对它们建立索引。我只想要带有组变量名称的列!这看起来很简单(并且在 R 的 ddply 中自动完成)，但经过大量谷歌搜索后我什么也没想到。有没有人有一个简单的解决方案？

Answer 1

默认情况下，分组变量将转换为索引。您可以使用 grouped.reset_index() 将索引更改为列。

我的第二个建议是在 groupby 调用中使用 as_index=False 指定这一点，在这种情况下，使用 apply 似乎无法按预期工作(但在使用时确实有效) 聚合)