Python Pandas - 选择行和列组合的子集

pythonserver side programmingprogramming更新于 2025/11/23 7:40:17

要选择行和列的子集,请使用 loc。使用索引运算符,即方括号,并在位置设置条件。

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

首先,将数据从 CSV 文件加载到 Pandas DataFrame −

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

选择行和列组合的子集。右列显示您想要显示的列,即此处的 Cars 列 −

dataFrame.loc[dataFrame["Units"] > 100, "Car"]

示例

以下是代码 −

import pandas as pd

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

# 选择行的子集
print("\n选择单位数超过 100 的汽车:\n",dataFrame[dataFrame["Units"] > 100])

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

# 选择行和列组合的子集
# 右列显示要显示的列,即此处的 Cars 列
res2 = dataFrame.loc[dataFrame["Units"] > 100, "Car"]

# 显示子集
print("\nSubset...\n",res2)

输出

这将产生以下输出 −

正在读取 CSV 文件...
       Car   Reg_Price   Units
0      BMW        2500     100
1    Lexus        3500      80
2     Audi        2500     120
3   Jaguar        2000      70
4  Mustang        2500     110

选择单位数超过 100 的汽车:
       Car   Reg_Price   Units
2     Audi        2500     120
4  Mustang        2500     110

仅显示两列:
    Reg_Price   Units
0        2500     100
1        3500      80
2        2500     120
3        2000      70
4        2500     110

Subset...
2 Audi
4 Mustang
Name: Car, dtype: object

相关文章


有用资源