根据列值从 Pandas DataFrame 中选择行
pandasserver side programmingprogramming更新于 2025/9/8 13:52:17
要根据列值从 DataFrame 中选择行，我们可以采取以下步骤 −
创建一个二维、大小可变、可能异构的表格数据 df。
打印输入 DataFrame。
使用 df.loc[df["x"]==2] 在 x==2 时打印 DataFrame。
类似地，在 (x >= 2) 和 (x < 2) 时打印 DataFrame。
示例
import pandas as pd
df = pd.DataFrame(
{
"x": [5, 2, 1, 9],
"y": [4, 1, 5, 10],
"z": [4, 1, 5, 0]
}
)
print "Given DataFrame is:
", df
print "When column x value == 2:
", df.loc[df["x"] == 2]
print "When column x value >= 2:
", df.loc[df["x"] >= 2]
print "When column x value < 2:
", df.loc[df["x"] < 2]
输出
Given DataFrame is: x y z 0 5 4 4 1 2 1 1 2 1 5 5 3 9 10 0 When column x value == 2: x y z 1 2 1 1 When column x value >= 2: x y z 0 5 4 4 1 2 1 1 3 9 10 0 When column x value < 2: x y z 2 1 5 5
