主页 / coding / 问题 / 79255967
Accepted
Shawn Zhang
Asked: 2024-12-06 03:15:59 +0800 CST

如何根据具有相同值的其他列创建增加值的列[重复]

  • 772

全部

我有如下数据框

df = pd.DataFrame.from_dict({'A':[1,1,2], 'B':[10,20,14], "C":[30,40,20]})

在此处输入图片描述

例如，如何创建一个newCol依赖于列的新列A。如果的值A保持不变，则将用一个增加的整数填充。如果这里newCol的值有所不同，那么的值将重置并再次开始增加。A2newCol

谢谢。

在此处输入图片描述

python

1 个回答

  1. Best Answer

    代码

    grp = df['A'].ne(df['A'].shift()).cumsum()
df['newCol'] = df['A'].groupby(grp).cumcount().add(1)

    数据：

       A   B   C  newCol
0  1  10  30       1
1  1  20  40       2
2  2  14  20       1
    • 2

相关问题