python DataFrame筛选符合特定条件的行

1. 获得a列中值为1的行

data[data['a'].isin([1])]

2. 获得a列中值为1或2的行

data[data['a'].isin([1,2])]

3. 获得a列中值大于1、小于2的行

data[(data['a']<2)&(data['a']>1)]

同时，Pandas也提供了query()方法来对DataFrame进行过滤。

DataFrame.query(expr, inplace=False, **kwargs)

参数
expr：引用字符串形式的表达式以过滤数据。
inplace：如果该值为True, 它将在原始DataFrame中进行更改。

1. 获得a列中值为1的行

data.query('a==1')

2. 获得a列中值为1或2的行

data.query('a==1 | a==2')

3. 获得a列中值大于1、小于2的行

data.query('2>a>1')