python - 将计数添加到新列 pandas python 3

Question

我正在尝试删除重复行并使用 pandas 写入出现次数。以下声明是我尝试过的:

createModel['count'] = createModel.groupby(createModel.columns.tolist(),as_index=False).size()
createModel.to_csv(r"test1.csv",index=False,header =True,sep="\t",encoding="utf-16")
createModel.head(10)

但我收到错误:TypeError:插入列的索引与框架索引不兼容


我知道这是因为我添加了count。如果我删除它然后尝试保存文件，我只能保护 count。

请告诉我如何保存完整的数据帧而不重复，并添加一列作为 count 表示该行出现的次数。

Answer 1

使用transform对于新列，但必须在 [] 中的 groupby 之后添加一列值:

cols = createModel.columns.tolist()
#another solution, thanks @jpp
#cols = list(createModel)
createModel['count'] = createModel.groupby(cols)[cols[0]].transform('size')

如果需要删除重复值，则需要聚合值并添加 reset_index :

createModel = createModel.groupby(cols).size().reset_index(name='count')