pandas中query()用法小結

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

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

一、pandas庫簡介

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

二、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ù)據(jù)。

示例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ù)據(jù)操作。

示例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列分組，并在每個組內(nèi)篩選B列的最大值
grouped_df = df.groupby('A').apply(lambda x: x.query('B == B.max()'))
print(grouped_df)

輸出：

A B C
A
1 1 1 6 q
2 3 2 7 s
3 5 3 8 q
4 7 4 9 s

五、總結

query()方法是pandas庫中一個強大而靈活的工具，它允許你使用易讀的字符串表達式來篩選DataFrame中的數(shù)據(jù)。通過結合邏輯運算符和字符串方法，你可以執(zhí)行復雜的數(shù)據(jù)篩選操作。然而，在處理大型數(shù)據(jù)集時，你應該注意性能問題，并考慮使用其他篩選方法。

到此這篇關于pandas中query()用法小結的文章就介紹到這了,更多相關pandas query()內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: