- 获得a列中值为1的行
data[data['a'].isin([1])]
- 获得a列中值为1或2的行
data[data['a'].isin([1,2])]
- 获得a列中值大于1、小于2的行
data[(data['a']<2)&(data['a']>1)]
同时,Pandas也提供了query()方法来对DataFrame进行过滤。
DataFrame.query(expr, inplace=False, **kwargs)
参数
expr:引用字符串形式的表达式以过滤数据。
inplace:如果该值为True, 它将在原始DataFrame中进行更改。
- 获得a列中值为1的行
data.query('a==1')
- 获得a列中值为1或2的行
data.query('a==1 | a==2')
- 获得a列中值大于1、小于2的行
data.query('2>a>1')