小编Rob*_*bin的帖子

我的目标是使用agg函数聚合 pandas DataFrameGroupBy 对象。

为了做到这一点，我正在生成一个字典，我将使用 dict unpacking through 将其解压为 kwargs **dict。该字典需要包含新列名作为键和元组作为值。元组的第一个值是列名称，该列名称被压缩为一个系列，并作为 的输入提供给第二个值lambda series: ...。

agg_dict = {
   f"{cat_name}_count": ('movement_state', lambda series: series.value_counts()[cat_name]) 
   for cat_name in ml_data['category_column'].cat.categories
}

# Aggregating
agg_ml_data = ml_data.groupby(['col1', 'col2']).agg(**agg_dict)

现在实际发生的事情对我来说有点奇怪。

假设：

ml_data['category_column'].cat.categories

Index(['cat1', 'cat2', 'cat3'], dtype='object')

一组的正确值计数是

one_group['category_column'].value_counts()

     | category_column
cat1 | 2
cat2 | 9
cat3 | 6

一组的预期输出：

一组的实际输出

不知何故，python 对 lambda 函数执行的 dict 理解并不像预期的那样，并且cat3在索引时仅使用最后一个类别值series.value_counts()[cat_name] …