home

หน้าแรก
 Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

Python - เลือกหลายคอลัมน์จาก Pandas DataFrame

บทนำ

ใช้วงเล็บเหลี่ยมคู่ dataFrame[['col1', 'col2']] เลือกหลายคอลัมน์พร้อมกัน

ขั้นตอน

  1. นำเข้า pandas
  2. อ่าน SalesData.csv
  3. เลือกคอลัมน์ Reg_Price และ Units: 
    dataFrame[['Reg_Price', 'Units']]

โค้ดเต็ม

import pandas as pd

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")
print("\nReading the CSV file...\n", dataFrame)

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

ผลลัพธ์

ได้ DataFrame เฉพาะ 2 คอลัมน์ Reg_Price และ Units