根据多列之间的差异过滤数据框

Question

我正在处理以下数据框，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 或者甚至没有增加，我们需要删除该名称。

期望的输出：

name

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

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

Answer 1

subset = df[df.filter(like='val_').T.diff().gt(10).any()]

输出：

>>> 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)]

输出：

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