gpt4 book ai didi

python-3.x - pandas 列出相同的索引

转载 作者:行者123 更新时间:2023-12-02 01:31:06 24 4
gpt4 key购买 nike

如果一个表连续 3 次具有相同的索引,我希望它为我获取这个数据帧。

示例

index  var1
1 a
2 b
2 c
2 d
3 e
2 f
5 g
2 f

代码之后

预期输出

index  var1
2 b
2 c
2 d

最佳答案

一种选择是在差异索引上分割数据帧,检查每个 block 的大小并过滤掉大小小于阈值的 block ,然后重新组合它们:

import pandas as pd
import numpy as np
diff_indices = np.flatnonzero(df['index'].diff().ne(0))

diff_indices
# array([0, 1, 4, 5, 6, 7], dtype=int32)

pd.concat([chunk for chunk in np.split(df, diff_indices) if len(chunk) >= 3])
index var1
1 2 b
2 2 c
3 2 d

关于python-3.x - pandas 列出相同的索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/73264724/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com