如果第一列中的元素等于前一行，则增加第二列值

Question

有与此类似的问题，但我真正要问的有点不同。

我想知道是否有一种方法可以在没有 for 循环（使用地图或柱状计算）的情况下实现以下代码（如果可能）或最快的方法。

我有一个包含 m 行（>1E7）和 n 列的 DataFrame（df）。第 j+1 列以全 1 或 0 开始。

for i in range(len(df)):

    if df.iloc[i, j] == df.iloc[i-1, j]: 

        df.iloc[i, j+1] = df.iloc[i-1, j+1]+1

因此示例输出将如下所示：

    ... j j+1 ...

  0 ... 3  1  ...

  1 ... 4  1  ...

  2 ... 4  2  ...

  3 ... 4  3  ...

  4 ... 6  1  ...

  5 ... 6  2  ...

  6 ... 7  1  ...

Answer 1

肯定有一些问题可以回答这个问题：s = df.iloc[:,j]blocks = s.ne(s.shift()).cumsum()df.iloc[:,j+1]= s.groupby(blocks).cumcount() + 1输出：   ...  j  j+1   ...0  ...  3    1   ...1  ...  4    1   ...2  ...  4    2   ...3  ...  4    3   ...4  ...  6    1   ...5  ...  6    2   ...6  ...  7    1   ...

Answer 2

听起来这就是你所追求的。df['j+1'] = df.groupby('j').cumcount() + 1输出：    j   j+10   3   11   4   12   4   23   4   34   6   15   6   26   7   1