使用值列表從 Pandas DataFrame 中選擇行

若要基於輸入值從 Pandas DataFrame 中選擇行，我們可以使用**isIn()** 方法。

步驟

• 建立一個二維的、大小可變的、可能的異構表格資料df。

• 列印輸入 DataFrame。

• 為選擇行建立一個值列表。

• 使用給定值列印選定的行。

• 接下來，列印未選中的行。

示例

 即時演示

import pandas as pd

df = pd.DataFrame(
   {
      "x": [5, 2, 1, 9],
      "y": [4, 1, 5, 10],
      "z": [4, 1, 5, 0]
   }
)

print "Input DataFrame:
", df
values = [1, 2]
print "Selected Rows:
", df[df['x'].isin(values)]
print "Uselected Rows:
", df[~df['x'].isin(values)]

輸出

Input DataFrame:
   x  y  z
0  5  4  4
1  2  1  1
2  1  5  5
3  9 10  0

Selected Rows:
   x  y  z
1  2  1  1
2  1  5  5

Unselected Rows:
   x  y  z
0  5  4  4
3  9 10  0

Rishikesh Kumar Rishi

更新於： 2021-08-30

2K+ 瀏覽

開啟你的 職業生涯

完成課程獲得認證

開始