python - 我想根据 pandas 中的另一列数据添加新列

Question

我有多个 csv 文件，之后我将其合并在一起，以便识别所有合并的 csv 文件中的各个 csv 数据，我希望在 pandas 中创建一个新列，其中新列应称为串行。

我想要在 pandas 中添加一个新的列序列，它应该根据序列列中的数据进行编号(例如，对于 csv 中的每个新列，111111111,2222222222,33333333)。我还 append 了 csv 文件的快照.

Sequence Number
1
2
3
4
5
1
2
1
2
3
4

我想要这样的输出-

Serial  Sequence Number
1   1
1   2
1   3
1   4
1   5
2   1
2   2
3   1
3   2
3   3
3   4

Answer 1

使用DataFrame.insert对于第一个位置的列，填充了 bool 掩码，以便通过 1 与 Series.eq 进行比较(==) 和 Series.cumsum 的累积和:

df.insert(0, 'Serial', df['Sequence Number'].eq(1).cumsum())
print (df)
    Serial  Sequence Number
0        1                1
1        1                2
2        1                3
3        1                4
4        1                5
5        2                1
6        2                2
7        3                1
8        3                2
9        3                3
10       3                4