Python Pandas – 从数据框中选择子集行

要选择一部分行，请使用条件并获取数据。

假设以下是在Microsoft Excel中打开的CSV文件内容−

首先，从CSV文件中加载数据到Pandas DataFrame中−

dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv")

假设我们想要“Units”大于100的汽车记录，即行子集。 为此，请使用−

dataFrame[dataFrame["Units"] > 100]

现在，让我们说我们想要“Reg_Price”小于100的汽车记录，即行子集。 为此，请使用−

dataFrame[dataFrame["Reg_Price"] < 3000]

范例

以下是代码−

import pandas as pd

# 从CSV文件中加载数据到Pandas DataFrame
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv")
print("\n读取CSV文件...\n",dataFrame)

# 显示两列
res2 = dataFrame[['Reg_Price','Units']];
print("\n显示两列：\n",res2)

# 选择部分行
print("\n选择单位大于100的汽车：\n",dataFrame[dataFrame["Units"] > 100])

# 选择部分行
print("\n选择Reg_Price小于3000的车：\n",dataFrame[dataFrame["Reg_Price"] < 3000])

输出

这将产生以下输出−

读取CSV文件...
       汽车   Reg_Price   单位
0      宝马        2500     100
1    雷克萨斯        3500      80
2     奥迪         2500     120
3   豹子        2000      70
4  野马         2500     110

仅显示一列汽车：
    Reg_Price   Units
0        2500     100
1        3500      80
2        2500     120
3        2000      70
4        2500     110
Name: Car, dtype: object

选择单位大于100的汽车：
       汽车   Reg_Price   单位
2     奥迪         2500     120
4  野马         2500     110

选择Reg_Price小于3000的车：
       汽车   Reg_Price   Units
0      宝马        2500     100
2     奥迪         2500     120
3   豹子        2000      70
4  野马         2500     110