按组查找第一次重复出现至少n次的值
创始人
2024-09-02 19:39:54
0

可以使用 pandas 库中的 groupby 和 rolling 函数来解决。首先按照分组条件进行分组,然后在每个组内使用 rolling 函数计算每个值在前 n-1 个值中是否与自己相同,得到一个布尔型的 rolling 对象,再使用 all 函数将其聚合成一个布尔值,表示前 n-1 个值是否都与当前值相同。最后再使用 shift 函数将这个布尔型 Series 往后移动一个位置,相邻两个位置进行比较,如果相邻两个位置都为 True,表示当前值在前 n 个值中出现了至少 n 次,则返回当前值的行号。

代码示例:

import pandas as pd

# 构造测试数据
df = pd.DataFrame({
    'group': ['A', 'A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'B', 'B'], 
    'value': ['a', 'b', 'c', 'a', 'a', 'd', 'a', 'b', 'b', 'b', 'a', 'a']
})

n = 3  # 设定连续出现次数的阈值

# 按组进行分组并使用 rolling 函数计算每个值在前 n-1 个值中是否与自己相同
rolling_obj = df.groupby('group').value.rolling(n-1).apply(lambda x: (x == x[-1]).all(), raw=True)

# 使用 shift 函数将这个布尔型 Series 往后移动一个位置
shifted_rolling_obj = rolling_obj.groupby('group').shift(periods=1)

# 如果相邻两个位置都为 True,表示当前值在前 n 个值中出现了至少 n 次,则返回当前值的行号
result = (rolling_obj & shifted_rolling_obj).reset_index(level=0, drop=True).index[0]

相关内容

热门资讯

iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
安卓系统怎么连不上carlif... 安卓系统无法连接CarLife的原因及解决方法随着智能手机的普及,CarLife这一车载互联功能为驾...
iphone系统与安卓系统更新... 最近是不是你也遇到了这样的烦恼?手机更新系统总是失败,急得你团团转。别急,今天就来给你揭秘为什么iP...
oppo手机安卓系统换成苹果系... OPPO手机安卓系统换成苹果系统:现实吗?如何操作?随着智能手机市场的不断发展,用户对于手机系统的需...
安卓平板改windows 系统... 你有没有想过,你的安卓平板电脑是不是也能变身成Windows系统的超级英雄呢?想象在同一个设备上,你...
安卓系统上滑按键,便捷生活与高... 你有没有发现,现在手机屏幕越来越大,操作起来却越来越方便了呢?这都得归功于安卓系统上的那些神奇的上滑...
安卓系统连接耳机模式,蓝牙、有... 亲爱的手机控们,你们有没有遇到过这种情况:手机突然变成了“耳机模式”,明明耳机没插,声音却只从耳机孔...
安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...
希沃系统怎么装安卓系统,解锁更... 亲爱的读者们,你是否也像我一样,对希沃一体机上的安卓系统充满了好奇呢?想象在教室里,你的希沃一体机不...
安装了Anaconda之后找不... 在安装Anaconda后,如果找不到Jupyter Notebook,可以尝试以下解决方法:检查环境...