熊猫：如何检查同一数据帧中列之间的值匹配？

Question

我是一名完整的编码新手，并一直在试验熊猫。这是我的第一篇文章。提前感谢您的帮助！

我想删除cat1与dog1或dog2不匹配的所有行。它不必匹配两者，只需要匹配一个或另一个。

randint

我希望最终结果如下：

   cat1   dog1   dog2
0   red   red    blue  
1   red   green  blue  
2   blue  red    blue  
3   blue  blue   green  
4   red   green  blue

我该怎么做？

Answer 1

这很简单：

df.query('cat1 == dog1 or cat1 == dog2')