根据Pandas中的Row值归档列

时间:2018-01-11 08:17:07

标签: python pandas

我可以使用布尔系列基于列值过滤行。

import pandas as pd
import numpy as np
from numpy.random import randn
df = pd.DataFrame(randn(5,4), ['A', 'B', 'C', 'D', 'E'], ['W','X','Y','Z'])

>>> df[df['W'] < 0]
          W         X         Y         Z
A -1.080180 -0.512936  0.810030  0.135316
D -0.713363  0.376887 -0.334995  0.864555

如何基于布尔系列过滤行,如何根据列布尔系列过滤列?像这样的东西。

df[df.loc['A'] > 1] ##It's wrong syntax I know. 

1 个答案:

答案 0 :(得分:3)

您非常接近,需要loc :才能按条件选择所有行:

np.random.seed(45)
df = pd.DataFrame(np.random.rand(5,4), ['A', 'B', 'C', 'D', 'E'], ['W','X','Y','Z'])
print (df)
          W         X         Y         Z
A  0.989012  0.549545  0.281447  0.077290
B  0.444469  0.472808  0.048522  0.163324
C  0.115951  0.627392  0.856182  0.650102
D  0.990722  0.470351  0.618294  0.282667
E  0.976003  0.673068  0.440531  0.289687

df1 = df.loc[:, df.loc['A'] > .5]
print (df1)
          W         X
A  0.989012  0.549545
B  0.444469  0.472808
C  0.115951  0.627392
D  0.990722  0.470351
E  0.976003  0.673068