pandas dataframe 按特定列中的值序列过滤

Question

我有一个数据框

A B C

1 2 3

2 3 4

3 8 7

我只想获取 C 列中有 3,4 序列的行(在这种情况下 - 前两行)

这样做的最佳方式是什么？

Answer 1

您可以使用rolling 来处理任何模式的通用解决方案:

pat = np.asarray([3,4])
N = len(pat)

mask= (df['C'].rolling(window=N , min_periods=N)
              .apply(lambda x: (x==pat).all(), raw=True)
              .mask(lambda x: x == 0) 
              .bfill(limit=N-1)
              .fillna(0)
              .astype(bool))

df = df[mask]
print (df)
   A  B  C
0  1  2  3
1  2  3  4

解释:

• 使用rolling.apply和测试模式
• 用mask将0替换为NaN
• 使用 bfill 和 limit 来用前一个 NAN 值填充前一个值
• fillna NaN 到 0
• 最后由 astype 转换为 bool