我有一个熊猫数据框,我想根据一些规则对其进行过滤。例如,
假设我有一个数据帧df
,它有3列:n, A, B
。 A
列中可能有nan值。假设它如下:
n A B
0 2 4
1 5 7
2 nan 11
3 4 3
4 6 8
5 1 0
6 nan 4
我想在A
值比B
值更好的地方过滤它。你会怎么做?
我尝试了这个,当然没用。
indices = []
for i,row in df.iterrows():
if(pd.notna(row['A'])):
if(row['A']>row['B']):
indices.append(i)
df[df.index in indices]