利用query函数可筛选、查询数据
df.query(expr, inplace=False, **kwargs)
参数:
-expr:查询条件,表达式
-inplace :是否替换原数据,默认为false
1、基本使用
import pandas as pd
df = pd.DataFrame(
{'A': ['e', 'd', 'c', 'b', 'a'],
'B': ['f', 'b', 'c', 'd', 'e'],
'C': range(0, 10, 2),
'D': range(10, 0, -2),
'E.E': range(10, 5, -1)})
我们现在选取 A列字母出现在B列 的所有行。先看两种常见写法
>>> df[df['A'].isin(df['B'])]
A B C D E.E
0 e f 0 10 10
1 d b 2 8 9
2 c c 4 6 8
3 b d 6 4 7
>>> df.loc[df['A'].isin(df['B'])]
A B C D E.E
0 e f 0 10 10
1 d b 2 8 9
2 c c 4 6 8
3 b d 6 4 7
下面使用 query() 来实现。
>>> df.query("A in B")
A B C D E.E
0 e f 0 10 10
1 d b 2 8 9
2 c c 4 6 8
3 b d 6 4 7
可以看到使用 query 后的代码简洁易懂,并且它对于内存的消耗也更小。
2、多条件查询
选取 A列字母出现在B列,并且C列小于D列 的所有行。
>>> df.query('A in B and C < D')
A B C D E.E
0 e f 0 10 10
1 d b 2 8 9
2 c c 4 6 8
这里 and 也可以用 & 表示。
3、引用变量
表达式中也可以使用外部定义的变量,在变量名前用@标明。
>>> number = 5
>>> df.query('A in B & C > @number')
A B C D E.E
3 b d 6 4 7
4、索引选取
选取 A列字母出现在B列,并且索引大于2 的所有行。
>>> df.query('A in B and index > 2')
A B C D E.E
3 b d 6 4 7
多索引选取
创建一个两层索引的 DataFrame。
>>> import numpy as np
>>> colors = ['yellow']*3 + ['red']*2
>>> rank = [str(i) for i in range(5)]
>>> index = pd.MultiIndex.from_arrays([colors, rank], names=['color', 'rank'])
>>> df = pd.DataFrame(np.arange(10).reshape(5, 2),columns=['A', 'B'] , index=index)
>>> df = pd.DataFrame(np.arange(10).reshape(5, 2),columns=['A', 'B'] , index=index)
>>> df
A B
color rank
yellow 0 0 1
1 2 3
2 4 5
red 3 6 7
4 8 9
1.当有多层索引有名称时,通过索引名称直接选取。
>>> df.query("color == 'red'")
A B
color rank
red 3 6 7
4 8 9
2.当有多层索引无名时,通过索引级别来选取。
>>> df.index.names = [None, None]
>>> df.query("ilevel_0 == 'red'")
A B
red 3 6 7
4 8 9
>>> df.query("ilevel_1 == '4'")
A B
red 4 8 9