Pandas：获取连续具有相似值的行

Question

我有一个像这样的数据框..

      id     time    status
      --     ----    ------
      a       1        T
      a       2        F
      b       1        T
      b       2        T
      a       3        T
      a       4        T
      b       3        F
      b       4        T
      b       5        T

我想获取在特定阈值次数内连续具有状态“T”的 ID（在本例中为 2）。

因此，获取的行将是...

    id   time   status
    --   ----   ------
    b      1      T
    b      2      T
    a      3      T
    a      4      T
    b      4      T
    b      5      T

我可以想到一个迭代解决方案。 我正在寻找的是更像 Pandas/sql 的东西。 我想通过一个订单id ，然后time跟着一群由第一id ，然后status应该工作，但我想是肯定的。

Answer 1

通过Series.eq比较T值并使用Series.shift和Series.cumsum计算连续值，通过Series.value_counts和Series.map计数到原始值 - 获取每个连续组的计数。 然后通过Series.ge进行比较，并通过boolean indexing链进行最后过滤，这两个掩码都按位AND ：

N = 2
m1 = df['status'].eq('T')
g = df['status'].ne(df['status'].shift()).cumsum()
m2 = g.map(g.value_counts()).ge(N)

df = df[m1 & m2]
print (df)
  id  time status
2  b     1      T
3  b     2      T
4  a     3      T
5  a     4      T
7  b     4      T
8  b     5      T

详情：

print (df.assign(m1=m1, g=g, counts=g.map(g.value_counts()), m2=m2))
  id  time status     m1  g  counts     m2
0  a     1      T   True  1       1  False
1  a     2      F  False  2       1  False
2  b     1      T   True  3       4   True
3  b     2      T   True  3       4   True
4  a     3      T   True  3       4   True
5  a     4      T   True  3       4   True
6  b     3      F  False  4       1  False
7  b     4      T   True  5       2   True
8  b     5      T   True  5       2   True