在使用 Pandas 的 `query` 方法进行数据筛选时，如果列名包含空格或其他非标准字符，用户可能会遇到 `KeyError`。本文将深入探讨 `query` 方法的工作原理，解释为何此类列名会导致错误，并提供使用反引号（“ ` “）引用这些列名的正确解决方案，同时对比 `query` 和 `.loc` 方法的使用场景，帮助读者更高效、准确地操作 DataFrame。
Pandas query 方法简介
Pandas 的 DataFrame.query() 方法提供了一种使用布尔表达式字符串来筛选 DataFrame 行的便捷方式。它允许用户以类似 SQL 的语法编写复杂的查询条件，这在某些情况下比传统的布尔索引（如 df.loc[]）更具可读性，尤其是在涉及多个条件时。
例如，如果我们有一个名为 df 的 DataFrame，并且想要筛选出 Speed 列值小于或等于 10 的行，可以使用以下简洁的 query 表达式：
import pandas as pd# 创建一个示例DataFramedata = { 'Name': ['Pikachu', 'Charmander', 'Squirtle', 'Bulbasaur', 'Jigglypuff'], 'Type 1': ['Electric', 'Fire', 'Water', 'Grass', 'Normal'], 'Type 2': [None, None, None, 'Poison', 'Fairy'], 'Speed': [90, 65, 43, 45, 20], 'HP': [35, 39, 44, 45, 115]}df = pd.DataFrame(data)# 使用 query 筛选 Speed <= 10 的行# 注意：在我们的示例数据中，没有 Speed <= 10 的行，所以会返回空DataFramedf.query("Speed <= 10")
理解 KeyError：列名中的空格问题
当列名是有效的 Python 变量名（即不包含空格、连字符等特殊字符，且不以数字开头）时，query 方法可以无缝地工作。然而，当列名包含空格时，例如 “Type 1” 或 “Type 2″，直接在 query 字符串中使用这些列名会导致 KeyError。
让我们看一个导致错误的例子：
# 尝试使用 query 筛选 'Type 1' 为 'Fire' 且 'Type 2' 为 'Flying' 的行# 假设df中存在这样的列和数据# df.query("'Type 1' == 'Fire' and 'Type 2' == 'Flying'")# 上述代码会引发 KeyError
这个 KeyError 的发生是因为 query 方法在解析字符串表达式时，会尝试将 Type 1 和 Type 2 解释为 Python 变量。由于它们不是有效的 Python 变量名（因为包含了空格），解析器无法正确识别它们作为 DataFrame 的列名，从而导致找不到对应的“键”，抛出 KeyError。
相比之下，使用传统的布尔索引 .loc 方法时，我们通过 df[‘Column Name’] 的方式明确指定列名，Pandas 会直接在 DataFrame 的列索引中查找，因此即使列名包含空格也能正常工作：
# 使用 .loc 方法进行相同条件的筛选，这是有效的df.loc[(df['Type 1'] == "Fire") & (df['Type 2'] == "Flying")]
解决方案：使用反引号引用列名
为了解决 query 方法中列名包含空格导致的 KeyError，Pandas 提供了使用反引号（`）来引用这些列名的机制。反引号告诉 query 解析器，其内部的字符串应该被视为一个字面量的列名，而不是一个 Python 变量。
以下是使用反引号修正后的 query 表达式：
# 修正后的 query 表达式，使用反引号引用含空格的列名df.query("`Type 1` == 'Fire' and `Type 2` == 'Flying'")
通过在 Type 1 和 Type 2 周围加上反引号，query 方法就能正确地识别这些列名并执行筛选操作。
示例代码：
为了更清晰地演示，我们使用一个包含“Type 1”和“Type 2”列的 DataFrame：
import pandas as pd# 创建一个包含含空格列名的示例DataFramedata = { 'Name': ['Charizard', 'Moltres', 'Ho-oh', 'Fletchinder', 'Talonflame', 'Pikachu'], 'Type 1': ['Fire', 'Fire', 'Fire', 'Fire', 'Fire', 'Electric'], 'Type 2': ['Flying', 'Flying', 'Flying', 'Flying', 'Flying', None], 'Total': [534, 580, 680, 382, 499, 320], 'Speed': [100, 90, 90, 84, 126, 90]}pokemon_df = pd.DataFrame(data)print("原始 DataFrame:")print(pokemon_df)print("n" + "="*30 + "n")# 错误示范 (会引发 KeyError)# try:# print("尝试使用错误的 query 语法 (将引发 KeyError):")# pokemon_df.query("'Type 1' == 'Fire' and 'Type 2' == 'Flying'")# except KeyError as e:# print(f"捕获到 KeyError: {e}")# print("n" + "="*30 + "n")# 正确示范：使用反引号引用含空格的列名print("使用反引号的正确 query 语法:")result_query = pokemon_df.query("`Type 1` == 'Fire' and `Type 2` == 'Flying'")print(result_query)print("n" + "="*30 + "n")# 对比：使用 .loc 方法实现相同功能print("使用 .loc 方法实现相同功能:")result_loc = pokemon_df.loc[(pokemon_df['Type 1'] == "Fire") & (pokemon_df['Type 2'] == "Flying")]print(result_loc)
何时使用 query 与 .loc
query 方法的优势：可读性： 对于复杂的筛选条件，query 字符串通常比嵌套的布尔索引表达式更易读。简洁性： 避免了重复的 df[…] 引用，使代码更紧凑。性能： 对于大型 DataFrame，query 在某些情况下可以利用 NumExpr 库进行优化，从而提供比纯 Python 布尔索引更快的性能。.loc 方法的优势：直接性： 直接通过列名进行索引，不会有解析字符串表达式的额外步骤，对于简单的筛选条件可能更直观。灵活性： .loc 不仅用于布尔筛选，还可以用于基于标签或布尔数组的行/列选择和修改，功能更全面。无列名限制： .loc 可以直接处理任何包含特殊字符的列名，无需额外的引用符号。
选择建议：
如果你的筛选条件较为复杂，并且希望代码更具可读性，同时列名符合 Python 变量命名规范或可以方便地使用反引号引用，query 是一个很好的选择。如果你的筛选条件简单，或者列名包含大量需要引用的特殊字符，或者你需要进行更广泛的行/列选择和修改，那么 .loc 可能是更直接和灵活的选择。
总结
Pandas 的 query 方法是一个强大的数据筛选工具，但了解其处理列名的方式至关重要。当列名包含空格或特殊字符时，务必使用反引号（`）将其包裹起来，以确保 query 方法能够正确识别并执行操作。掌握这一技巧，将使你在使用 query 方法时更加得心应手，编写出更健壮、更专业的 Pandas 代码。
以上就是Pandas query 方法深度解析：处理含空格列名的 KeyError的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1382195.html
微信扫一扫 
支付宝扫一扫 
