用条件过滤熊猫 df

时间:2021-07-21 11:59:43

标签: python pandas dataframe filter

我有一只熊猫 df:

df
                          DEtoDK         DKtoDE         
                            self other     self    other
2021-01-01 00:00:00+00:00    0.0   NaN  2230.08      NaN
2021-01-01 01:00:00+00:00    0.0   0.0  1887.72  2230.08
2021-01-01 02:00:00+00:00    0.0   0.0  1821.33  1887.72
2021-01-01 03:00:00+00:00    0.0   0.0  1743.20  1821.33
2021-01-01 04:00:00+00:00    0.0   0.0  1720.78  1743.20
...                          ...   ...      ...      ...
2021-05-31 19:00:00+00:00    0.0   0.0   782.88   892.16
2021-05-31 20:00:00+00:00    0.0   0.0   872.96   782.88
2021-05-31 21:00:00+00:00    0.0   0.0  1165.36   872.96
2021-05-31 22:00:00+00:00    0.0   0.0  1418.32  1165.36
2021-05-31 23:00:00+00:00    0.0   0.0  1393.28  1418.32
[3624 rows x 4 columns]

如果 (DEtoDK, self)(DKtoDE, self) 值为 0,我想过滤此 df。为此,我使用以下条件:

df.loc[(df[('DEtoDK', 'self')].values == 0) | (df[('DKtoDE', 'self')].values == 0)]

这有效,但是当 df 没有任何 0 值时,我希望生成一个空数据帧,但是我收到了 KeyError。

df.loc[(df[('DEtoDK', 'self')].values == 'test') | (df[('DKtoDE', 'self')].values == 'test')]
KeyError: False

所以条件生成False,而不是空数组,因此熊猫无法定位。我该如何解决这种行为?

1 个答案:

答案 0 :(得分:3)

不需要values

df.loc[(df[('DEtoDK', 'self')] == 0) | (df[('DKtoDE', 'self')] == 0)]