Pandas 如果满足条件则复制行，并赋值

Question

我有一个如下所示的 Pandas 数据框，其中 A 列 是一系列字符串值，B 列 维护运行 列 A 中的值与前一个 列 A 中的值不同的总次数行。

每次 A 列的值发生变化时，我想复制前一行并为其分配 B 列的增量值。例如，对于上面的输入数据框，输出看起来喜欢：

关于如何有效地解决这个问题有什么想法吗？

Answer 1

通过 B 筛选最后重复的值，然后仅移动 B 并分配回去，删除最后一行并通过 concat 最后连接在一起并按索引排序：

df1 = (df[df['B'].ne(df['B'].shift(-1))]
         .assign(B = lambda x: x.B.shift(-1)).iloc[:-1].astype({'B':int}))

df = pd.concat([df, df1]).sort_index(ignore_index=True)
print (df)
    A  B
0   1  1
1   1  1
2   1  2
3  1b  2
4  1b  2
5  1b  2
6  1b  3
7   1  3

Pandas 如果满足条件则复制行，并赋值

Pandas duplicating row if condition met, and assigning value

python

numpy

shift

pandas