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

สร้าง Subset DataFrame โดยเลือกคอลัมน์ด้วยชื่อ (Column Name)

แนวคิด

วิธีง่ายที่สุดคือใช้ square brackets พร้อมชื่อคอลัมน์: df['column_name'] (ได้ Series) หรือ df[['col1', 'col2']] (ได้ DataFrame)

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

import pandas as pd

df = pd.DataFrame({
    "Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],
    "Opening_Stock": [300, 700, 1200, 1500],
    "Closing_Stock": [200, 500, 1000, 900]
})

print("DataFrame:\n", df)

# Subset คอลัมน์ Product (ได้ Series)
print("\nSubset คอลัมน์ Product (Series):\n", df['Product'])

# Subset หลายคอลัมน์ (ได้ DataFrame)
print("\nSubset คอลัมน์ Product และ Opening_Stock (DataFrame):\n", df[['Product', 'Opening_Stock']])

ผลลัพธ์

DataFrame:
  Closing_Stock  Opening_Stock     Product
0           200            300      SmartTV
1           500            700   ChromeCast
2          1000           1200      Speaker
3           900           1500     Earphone

Subset คอลัมน์ Product (Series):
0      SmartTV
1   ChromeCast
2      Speaker
3     Earphone
Name: Product, dtype: object

Subset คอลัมน์ Product และ Opening_Stock (DataFrame):
     Product  Opening_Stock
0     SmartTV            300
1  ChromeCast            700
2     Speaker           1200
3    Earphone           1500

เทคนิคเพิ่มเติม

# ใช้ filter() กรองชื่อคอลัมน์
df.filter(like='Stock')        # คอลัมน์ที่มี 'Stock'
df.filter(regex='^Open')       # เริ่มต้นด้วย 'Open'
df.filter(items=['Product'])   # รายการชื่อคอลัมน์

# ใช้ loc[] (label-based)
df.loc[:, 'Product']           # Series
df.loc[:, ['Product', 'Opening_Stock']]  # DataFrame