python - 具有增量数字的新列，其初始值基于不同的列值(pandas)

Question

我想为定义的行中具有相同值的行添加带有增量编号的列；

例如如果我想要这个 df

df=pd.DataFrame([['a','b'],['a','c'],['c','b']])

我想要第一列的增量数字。它应该看起来像这样

df=pd.DataFrame([['a','b',1],['a','c',2],['c','b',1]])

我找到了 sql 解决方案，但我正在使用 ipython/pandas。有人可以帮助我吗？

Answer 1

使用cumcount ，对于新列的名称，使用原始列的length:

print (len(df.columns))
2

df[len(df.columns)] = df.groupby(0).cumcount() + 1
print (df)
   0  1  2
0  a  b  1
1  a  c  2
2  c  b  1