使用值列表从 Pandas DataFrame 中选择行
要根据输入值从PandasDataFrame中选择行,我们可以使用该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: x y z 0 5 4 4 3 9 10 0