pandas中query()用法小結

更新時間：2024年03月21日 11:40:01 作者：高斯小哥

query()方法是一個功能強大的函數(shù),允許用戶通過字符串表達式來篩選DataFrame中的數(shù)據,本文就來介紹一下pandas中query()用法,感興趣的可以了解一下

一、pandas庫簡介

pandas是Python中一個非常流行的數(shù)據處理庫，它提供了大量的數(shù)據結構（如Series和DataFrame）以及數(shù)據分析工具，使得數(shù)據處理變得既簡單又高效。在pandas中，query()方法是一個功能強大的函數(shù)，允許用戶通過字符串表達式來篩選DataFrame中的數(shù)據。

二、query()方法基礎

query()方法允許你使用字符串表達式來篩選DataFrame的行。這個表達式類似于你在Python中使用的常規(guī)表達式，但是它專門針對DataFrame的列名和值。

示例1：基本用法

import pandas as pd

# 創(chuàng)建一個示例DataFrame
data = {
    'A': [1, 2, 3, 4],
    'B': [5, 6, 7, 8],
    'C': ['p', 'q', 'r', 's']
}
df = pd.DataFrame(data)

# 使用query()方法篩選A列大于2的行
filtered_df = df.query('A > 2')
print(filtered_df)

輸出：

A B C
2 3 7 r
3 4 8 s

三、高級用法與技巧

query()方法不僅限于簡單的比較操作，你還可以使用邏輯運算符（如&、|）和更復雜的表達式來篩選數(shù)據。

示例2：使用邏輯運算符

import pandas as pd

# 創(chuàng)建一個示例DataFrame
data = {
    'A': [1, 2, 3, 4],
    'B': [5, 6, 7, 8],
    'C': ['p', 'q', 'r', 's']
}
df = pd.DataFrame(data)

# 篩選A列大于2且B列小于等于7的行
filtered_df = df.query('A > 2 and B <= 7')
print(filtered_df)

輸出：

A B C
2 3 7 r

示例3：使用字符串方法

import pandas as pd

# 創(chuàng)建一個示例DataFrame
data = {
    'A': [1, 2, 3, 4],
    'B': [5, 6, 7, 8],
    'C': ['p', 'qu', 'r', 's']
}
df = pd.DataFrame(data)

# 篩選C列以'q'開頭的行
filtered_df = df.query('C.str.startswith("q")')
print(filtered_df)

輸出：

A B C
1 2 6 qu

四、結合其他pandas功能

query()方法還可以與其他pandas功能（如groupby()、sort_values()等）結合使用，以執(zhí)行更復雜的數(shù)據操作。

示例4：結合groupby()

import pandas as pd

# 創(chuàng)建一個示例DataFrame
data = {
    'A': [1, 1, 2, 2, 3, 3, 4, 4],
    'B': [5, 6, 6, 7, 7, 8, 8, 9],
    'C': ['p', 'q', 'r', 's', 'p', 'q', 'r', 's']
}
df = pd.DataFrame(data)

# 按A列分組，并在每個組內篩選B列的最大值
grouped_df = df.groupby('A').apply(lambda x: x.query('B == B.max()'))
print(grouped_df)

輸出：