根据多列之间的差异过滤 Dataframe

Question

我正在研究以下 dataframe, df ：

name         val_1       val_2      val_3

AAA           20         25          30       
BBB           15         20          35
CCC           25         40          45
DDD           20         20          25

我只需要保留任何val列从前一个val列增加超过 10 的名称。 如果一列比前一列增加了不到 10 或者甚至没有增加，我们需要删除该名称。

所需的 output：

name

BBB #val3 increases by 15  
CCC #val2 increases by 15 

最聪明的做法是什么？ 任何建议，将不胜感激。 谢谢！

Answer 1

subset = df[df[['val_1', 'val_2', 'val_3']].diff().ge(10).any(axis=1)]

Output（假设val_3的AAA是 20 而不是 30）：

>>> subset
  name  val_1  val_2  val_3
1  BBB     15     20     35
2  CCC     25     40     45

Answer 2

我理解的方式是，当两行在所有列上的差异至少为 10 时，您希望保留这两行。

为此，您需要使用diff + ge + all构建一个掩码，并将掩码与其shift结合起来：

m = df.filter(like='val_').diff().ge(10).all(1)
out = df[m|m.shift(-1)]

output：

  name  val_1  val_2  val_3
1  BBB     15     20     35
2  CCC     25     40     45