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

เลือกคอลัมน์จาก Pandas DataFrame

แนวคิด

เลือกคอลัมน์จาก DataFrame ได้หลายวิธี วิธีพื้นฐานคือ square brackets df['column_name']

ตัวอย่างโค้ด

import pandas as pd

df = pd.DataFrame({
    "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
    "Units": [100, 150, 110, 80, 110, 90]
})

print("DataFrame:\n", df)

# เลือกคอลัมน์เดียว -> Series
car_series = df['Car']
print("\nคอลัมน์ Car (Series):\n", car_series)
print("Type:", type(car_series))

# เลือกหลายคอลัมน์ -> DataFrame (double brackets)
car_units = df[['Car', 'Units']]
print("\nคอลัมน์ Car และ Units (DataFrame):\n", car_units)
print("Type:", type(car_units))

# Attribute access (ทำงานได้ถ้าชื่อคอลัมน์เป็น valid identifier)
print("\ndf.Car:\n", df.Car)

ผลลัพธ์

DataFrame:
       Car  Units
0      BMW    100
1    Lexus    150
2     Audi    110
3  Mustang     80
4  Bentley    110
5   Jaguar     90

คอลัมน์ Car (Series):
0      BMW
1    Lexus
2     Audi
3  Mustang
4  Bentley
5   Jaguar
Name: Car, dtype: object

คอลัมน์ Car และ Units (DataFrame):
       Car  Units
0      BMW    100
1    Lexus    150
2     Audi    110
3  Mustang     80
4  Bentley    110
5   Jaguar     90

สรุปวิธีเลือกคอลัมน์

Syntaxคืนค่าใช้เมื่อ
df['col']Seriesคอลัมน์เดียว
df[['col1','col2']]DataFrameหลายคอลัมน์ หรือต้องการ DataFrame
df.loc[:, 'col']Seriesร่วมกับ row selection
df.iloc[:, 0]Seriesเลือกด้วย position
df.colSeriesชื่อคอลัมน์ไม่มี space/keyword