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

Question

我有一个来自大型问卷的 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 之前处理这些列名？

Answer 1

在 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了解更多详情。