首頁 >後端開發 >Python教學 >如何使用值列表對 Pandas DataFrame 進行子集化?

如何使用值列表對 Pandas DataFrame 進行子集化?

Barbara Streisand
Barbara Streisand原創
2024-12-21 10:43:10157瀏覽

How to Subset Pandas DataFrames Using a List of Values?

基於值列表對 Pandas 資料幀進行子集化

在資料分析中,通常需要根據預先定義的條件從資料幀中檢索特定行。 Pandas 提供了各種對資料幀進行子集化的方法,包括基於值清單選擇行的能力。

利用isin() 方法

根據值列表對Pandas 資料幀進行子集化,您可以使用isin() 方法,如圖所示:

import pandas as pd

# Create a Pandas dataframe
df = pd.DataFrame({'A': [5, 6, 3, 4], 'B': [1, 2, 3, 5]})

# Define a list of values to filter by
list_of_values = [3, 6]

# Subset dataframe based on the list
y = df[df['A'].isin(list_of_values)]

print(y)

輸出:

   A  B
1  6  2
2  3  3

isin()方法可讓您過濾指定列值與提供的清單中的任何值相符的行。

使用 ~

否定選擇在某些情況下,您可能需要根據值清單排除行。要實現此目的,您可以將~ 運算符與isin() 一起使用,如下所示:

import pandas as pd

# Create a Pandas dataframe
df = pd.DataFrame({'A': [5, 6, 3, 4], 'B': [1, 2, 3, 5]})

# Define a list of values to exclude
list_of_values = [3, 6]

# Subset dataframe excluding the list
z = df[~df['A'].isin(list_of_values)]

print(z)

輸出:

   A  B
0  5  1
3  4  5

~ 運算符否定選擇,確保行算符否定選擇顯示不在指定清單中的值。

以上是如何使用值列表對 Pandas DataFrame 進行子集化?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn