本篇文章带大家聊聊一个python pandas库的使用小技巧，介绍一下使用query()优雅查询的方法，希望对大家有所帮助！
对于 Pandas 根据条件获取指定数据，相信大家都能够轻松的写出相应代码，但是如果你还没用过 query，相信你会被它的简洁所折服！
常规用法
先创建一个 DataFrame。
import pandas as pddf = 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列 的所有行。先看两种常见写法。
立即学习“Python免费学习笔记（深入）”；
>>> df[df['A'].isin(df['B'])] A B C D E.E0 e f 0 10 101 d b 2 8 92 c c 4 6 83 b d 6 4 7>>> df.loc[df['A'].isin(df['B'])] A B C D E.E0 e f 0 10 101 d b 2 8 92 c c 4 6 83 b d 6 4 7
登录后复制
下面使用 query() 来实现。
>>> df.query("A in B") A B C D E.E0 e f 0 10 101 d b 2 8 92 c c 4 6 83 b d 6 4 7
登录后复制
可以看到使用 query 后的代码简洁易懂，并且它对于内存的消耗也更小。
多条件查询
选取 A列字母出现在B列，并且C列小于D列 的所有行。
>>> df.query('A in B and C 这里 and 也可以用 & 表示。
引用变量
表达式中也可以使用外部定义的变量，在变量名前用@标明。
>>> number = 5>>> df.query('A in B & C > @number') A B C D E.E3 b d 6 4 7
登录后复制
索引选取
选取 A列字母出现在B列，并且索引大于2 的所有行。
>>> df.query('A in B and index > 2') A B C D E.E3 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 Bcolor rank yellow 0 0 1 1 2 3 2 4 5red 3 6 7 4 8 9
登录后复制
1、当有多层索引有名称时，通过索引名称直接选取。
>>> df.query("color == 'red'") A Bcolor rank red 3 6 7 4 8 9
登录后复制
2、当有多层索引无名时，通过索引级别来选取。
>>> df.index.names = [None, None]>>> df.query("ilevel_0 == 'red'") A Bred 3 6 7 4 8 9>>> df.query("ilevel_1 == '4'") A Bred 4 8 9
登录后复制
特殊字符
对于列名中间有空格或运算符等其他特殊符号，需要使用反引号 ``。
>>> df.query('A == B | (C + 2 > `E.E`)') A B C D E.E2 c c 4 6 83 b d 6 4 74 a e 8 2 6
登录后复制
总的来说，query() 用法比较简单，可以快速上手，代码可读性也提高了不少。
【相关推荐：Python3视频教程 】
以上就是一文了解Python中如何使用query()进行优雅的查询的详细内容，更多请关注【创想鸟】其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至253000106@qq.com举报，一经查实，本站将立刻删除。
发布者：PHP中文网，转转请注明出处：https://www.chuangxiangniao.com/p/2245314.html
