Skip to main content
 首页 » 编程设计

python之分组数据帧上的列名输出到 CSV

2025年05月04日9sxdcgaq8080

我有一个来自大型问卷的 dataFrame,我通过聚合不同轴上的数据来生成摘要:

df.groupby(group_name).agg([np.mean, np.std, np.count_nonzero]) 

这会生成一列,其中包含我调查问卷中每个问题的均值、标准差和计数。分组的dataFrame中每一列的名称是一个元组(original_column_name, function_applied)

问题是,当我输出到 CSV(使用 to_csv())时,列名作为元组输出,即 ('gender', 'mean'), ('gender ', 'std') 理想情况下我想要像 gender_mean & gender_std

这样的东西

如何在输出到 CSV 之前处理这些列名?

请您参考如下方法:

在 pandas 0.8.1 中,试试这个:

group_df = df.groupby(group_name).agg([np.mean, np.std, np.count_nonzero]) 
group_df.rename(None, lambda coltuple: '_'.join(coltuple), False, True) 

参见 the DataFrame documentation了解更多详情。