如果列中的值等于另一个+5 max(在熊猫中),则保留行

时间:2020-04-01 14:54:43

标签: pandas

你好,

我有一个df,例如

col1 col2 col3
G1 2 2
G2 1 6
G3 2 20
G4 2 2
G5 1 100
G6 2 8
G7 2 4

我知道如何仅用

保留col2 == col3的行
tab2 = tab2[tab2['col2'] == tab2['col3']]

但是怎么说

tab2 = tab2 [tab2 ['col2'] ==直到+5 tab2 ['col3']]

这意味着如果col1中的值为1,而col2中的值为6,则我可以保留一行,因为1 + 5 = 6。 在这里我应该得到:

col1 col2 col3
G1 2 2
G2 1 6
G4 2 2
G7 2 4

0 个答案:

没有答案