使用值列表从 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+ 浏览

开启你的 职业生涯

完成课程获得认证

开始