根据列值从 Pandas DataFrame 中选择行

以下步骤可以根据列值从 DataFrame 中选择行 -

  • 创建一个二维、大小可变、可能为异构表格数据，df。

  • 打印输入 DataFrame。

  • 当 x==2 时，使用 df.loc[df["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

更新于: 2021 年 8 月 30 日

814 次浏览

开启你的职业生涯

完成课程获得认证

开始使用
广告