使用值列表从 Pandas DataFrame 中选择行
pandasserver side programmingprogramming更新于 2025/9/8 13:22:17
要根据输入值从 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
