按列（“租户”）分组并获取（“值”）列中最大连续 1

Question

我有一个如下所示的 df：

租户	价值
x	1
x	1
x	0
x	1
y	1
y	0

结果： 租户 X 应为 2，租户 y 应为 1

我正在尝试获取每组的最大连续值 1。如果值 1 之间有任何 0，则计数重新开始。 我是 python 新手，不知道从哪里开始。谢谢

Answer 1

您可以尝试：

def max_ones(col):
    return col.groupby(col.diff().ne(0).cumsum()).sum().max()

result = df.groupby("Tenant").Value.agg(max_ones)

如果 Value 列可以有 0 或 1 以外的其他值，您应该使用：

result = df.assign(Value=df.Value.eq(1)).groupby("Tenant").Value.agg(max_ones)