在 Python 中聚合具有选择性列的数据帧时添加标头

Question

我有一个包含三列的数据帧，即'Ts_id'、'doc_id'、'doc_type'，我发现计数仅聚合在'doc_id'和'doc_type'上。但是当我将聚合值写入 CSV 时，"Ts_id"也作为标题出现。即我现在收到四个标题，分别是'Ts_id'、'doc_id'、'doc_type'和'count''Ts_Id'列是空白且不必要的。有没有办法重命名列或在聚合后添加不同的标题？ 下面提供了这段代码：

dfnonreturns = pd.DataFrame(GuidedocdetailsNonReturn,columns=['Ts_id','doc_id','doc_type'])
dfnonreturnguidecount = dfnonreturns.groupby(['doc_id','doc_type']).agg(['count'])
dfnonreturnguidecount.to_csv('NonReturn_guideid_counts.csv',header=True)

Python 版本是 2.7

Answer 1

尝试以下操作：

dfnonreturns = pd.DataFrame(GuidedocdetailsNonReturn,columns=['Ts_id','doc_id','doc_type'])
dfnonreturnguidecount = dfnonreturns[['doc_id','doc_type']].groupby(['doc_id','doc_type']).agg(['count'])
dfnonreturnguidecount.to_csv('NonReturn_guideid_counts.csv',header=True)