Python Pandas - Chọn tập con hàng và cột kết hợp

Chọn tập con hàng và cột dùng loc. Dùng chỉ số vuông và điều kiện trong loc.

File CSV SalesData.csv.

Tải dữ liệu:

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

Chọn hàng có Units > 100 và chỉ cột Car:

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

Ví dụ

import pandas as pd 
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv") 
print("\nĐọc file CSV...\n", dataFrame) 
print("\nChọn xe Units > 100:\n", dataFrame[dataFrame["Units"] > 100]) 
res = dataFrame[['Reg_Price', 'Units']] 
print("\nChỉ 2 cột:\n", res) 
res2 = dataFrame.loc[dataFrame["Units"] > 100, "Car"] 
print("\nTập con...\n", res2)

Kết quả

Reading the CSV file... 
  Car  Reg_Price  Units 
0 BMW  2500  100 
1 Lexus  3500  80 
2 Audi  2500  120 
3 Jaguar  2000  70 
4 Mustang  2500  110 
Select cars with Units more than 100: 
  Car  Reg_Price  Units 
2 Audi  2500  120 
4 Mustang  2500  110 
Displaying only two columns: 
  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