使用值列表从 Pandas DataFrame 中选择行

Z时代
2024-01-10
分类：综合

要根据输入值从 Pandas DataFrame 中选择行，我们可以使用该isin()方法。

步骤

创建二维、大小可变、潜在异构的表格数据df。
打印输入数据帧。
创建用于选择行的值列表。
打印具有给定值的选定行。
接下来，打印未选择的行。

示例

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:\n", df
values = [1, 2]
print "Selected Rows:\n", df[df['x'].isin(values)]
print "Uselected Rows:\n", 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: