python.query()函数对DataFrame的选取

发表于 | 分类于

创建DataFrame:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import pandas as pd
d={
    'name':['a','n','c','d','e','f'],
    'Gender':['male','female','male','male','female','female'],
    'age':[23,24,24,22,21,20],
    'hight':[173,174,164,172,161,160],
    'weight1':[53,74,44,62,71,60],
    'weight2':[53,64,54,66,81,50]
}
df=pd.DataFrame(d)
df
Out[6]: 
  name  Gender  age  hight  weight1  weight2
0    a    male   23    173       53       53
1    n  female   24    174       74       64
2    c    male   24    164       44       54
3    d    male   22    172       62       66
4    e  female   21    161       71       81
5    f  female   20    160       60       50

一般来说如果进行行挑选，可以进行的操作是：

1
2
3
4
5
6
7
8
9
10
df[df.age==24]
Out[13]: 
  name  Gender  age  hight  weight1  weight2
1    n  female   24    174       74       64
2    c    male   24    164       44       54
df[(df.age==24 )&( df.hight ==174)]
Out[14]: 
  name  Gender  age  hight  weight1  weight2
1    n  female   24    174       74       64

但是如果用python.query函数，则更加有逻辑且代码更优雅

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
df.query("age==24")
Out[20]: 
  name  Gender  age  hight  weight1  weight2
1    n  female   24    174       74       64
2    c    male   24    164       44       54
df.query("age==24").query('hight==174')
Out[21]: 
  name  Gender  age  hight  weight1  weight2
1    n  female   24    174       74       64
Out[29]: 
  name Gender  age  hight  weight1  weight2
0    a   male   23    173       53       53
2    c   male   24    164       44       54
3    d   male   22    172       62       66
df.query('index > 2')
Out[47]: 
  name  Gender  age  hight  weight1  weight2
3    d    male   22    172       62       66
4    e  female   21    161       71       81
5    f  female   20    160       60       50
df.query('Gender =="male" and name =="a"')
Out[30]: 
  name Gender  age  hight  weight1  weight2
0    a   male   23    173       53       53
df.query('Gender =="male" and age<24')
Out[31]: 
  name Gender  age  hight  weight1  weight2
0    a   male   23    173       53       53
3    d   male   22    172       62       66

除此之外，query（）函数还可以进行不同列之间的值对比：

1
2
3
4
5
df.query('weight1 > weight2')
Out[22]: 
  name  Gender  age  hight  weight1  weight2
1    n  female   24    174       74       64
5    f  female   20    160       60       50

原文链接：https://www.jianshu.com/p/d174c6ed6d2e

原作者：柳叶刀与小鼠标

0%