如何使用值列表有效過濾 Pandas DataFrame？

使用值列表過濾 Pandas 資料幀

在資料操作任務中，根據值清單從 Pandas 資料幀中選擇特定行是常見的要求。本文示範如何有效率地實現此操作。

使用 isin() 方法

要選擇給定清單中存在指定列的值的行，可以使用 isin() 方法是一個簡單的解決方案。讓我們考慮以下Pandas 資料框：

df = pd.DataFrame({'A': [5,6,3,4], 'B': [1,2,3,5]})
print(df)

要擷取欄位「A」包含值3 或6 的行，我們可以使用：

list_of_values = [3, 6]
result = df[df['A'].isin(list_of_values)]
print(result)

此操作會產生符合的行「A」列值：

   A  B
1  6  2
2  3  3

排除帶 ~的值運算子

要排除清單中不存在「A」列值的行，~ 運算子可以與 isin() 結合使用。例如：

result = df[~df['A'].isin(list_of_values)]
print(result)

此操作排除 'A' 值為 3 或 6 的行：

   A  B
0  5  1
3  4  5

以上是如何使用值列表有效過濾 Pandas DataFrame？的詳細內容。更多資訊請關注PHP中文網其他相關文章！