Python - Chọn nhiều cột từ khung dữ liệu Pandas

Giả sử sau đây là nội dung của tệp CSV của chúng tôi được mở trong Microsoft Excel -

Đầu tiên, hãy tải dữ liệu từ tệp CSV vào Pandas DataFrame -

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

Để chọn nhiều bản ghi cột, hãy sử dụng dấu ngoặc vuông. Đề cập đến các cột trong dấu ngoặc và tìm nạp nhiều cột từ toàn bộ tập dữ liệu -

dataFrame[['Reg_Price','Units']]

Ví dụ

Sau đây là mã -

import pandas as pd

# Load data from a CSV file into a Pandas DataFrame:
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")
print("\nReading the CSV file...\n",dataFrame)

# displaying two columns
res = dataFrame[['Reg_Price','Units']];
print("\nDisplaying two columns : \n",res)

Đầu ra

Điều này sẽ tạo ra kết quả sau -

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

Displaying two columns :
    Reg_Price   Units
0        2500     100
1        3500      80
2        2500     120
3        2000      70
4        2500     110