如何有效地向分组的 Pandas DataFrame 添加顺序计数器列？

在没有回调的情况下将组上的顺序计数器列添加到 Pandas DataFrame

在 pandas 库中，可以使用比用于将顺序计数器列添加到分组数据帧的回调函数。 cumcount() 函数提供了一种替代方法。以下是如何实现它：

import pandas as pd

df = pd.DataFrame(
    columns="   index    c1    c2    v1 ".split(),
    data=[
        [0, "A", "X", 3],
        [1, "A", "X", 5],
        [2, "A", "Y", 7],
        [3, "A", "Y", 1],
        [4, "B", "X", 3],
        [5, "B", "X", 1],
        [6, "B", "X", 3],
        [7, "B", "Y", 1],
        [8, "C", "X", 7],
        [9, "C", "Y", 4],
        [10, "C", "Y", 1],
        [11, "C", "Y", 6],
    ],
).set_index("index", drop=True)

# Add a sequential counter column
df["seq"] = df.groupby(["c1", "c2"]).cumcount() + 1
print(df)

此代码片段按“c1”和“c2”对数据帧进行分组，然后将 cumcount() 函数应用于每个组以计算该组内的行数。结果是一个带有附加“seq”列的数据帧，其中包含每个组的顺序计数：

   c1 c2  v1  seq
0   A  X   3    1
1   A  X   5    2
2   A  Y   7    1
3   A  Y   1    2
4   B  X   3    1
5   B  X   1    2
6   B  X   3    3
7   B  Y   1    1
8   C  X   7    1
9   C  Y   4    1
10  C  Y   1    2
11  C  Y   6    3

这种方法消除了对回调函数的需要，并为添加顺序计数器列提供了更简洁、更高效的解决方案分组的 pandas 数据框。

以上是如何有效地向分组的 Pandas DataFrame 添加顺序计数器列？的详细内容。更多信息请关注PHP中文网其他相关文章！