如何检查相同连续数字的出现是否低于熊猫系列中的阈值

Question

我需要检查相同连续数字的出现是否低于某个阈值，例如最多两个相同的连续数字。

pd.Series(data=[-1, -1, 2, -2, 2, -2, 1, 1])  # True
pd.Series(data=[-1, -1, -1, 2, 2, -2, 1, 1])  # False

进一步检查：
仅允许数字 +1 和 -1 作为连续数字出现，最多出现两次。

pd.Series(data=[-1, 1, -2, 2, -2, 2, -1, 1])  # True
pd.Series(data=[1, 1, -2, 2, -2, 2, -1, 1])  # True
pd.Series(data=[-1, -1, 2, 2, -2, 1, 1, -2])  # False
pd.Series(data=[-1, 1, -2, -2, 1, -1, 2, -2]) # False

Answer 1

groupby.size 和 any 选项：

def check(s, N=2):
    return s.groupby(s.ne(s.shift()).cumsum()).size().gt(N).any()

或者使用 rolling 总和：

def check(s, N=2):
    return s.rolling(N+1).apply(lambda x: x.eq(x.iloc[0]).sum()>N).any()

check(pd.Series(data=[-1, -1, 2, -2, 2, -2, 1, 1]))
# False

check(pd.Series(data=[-1, -1, -1, 2, 2, -2, 1, 1]))
# True