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

Python Pandas – แสดงชื่อคอลัมน์ทั้งหมดของ DataFrame

ภาพรวม

ใช้ attribute .columns ของ DataFrame เพื่อดูชื่อคอลัมน์ทั้งหมด

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


import pandas as pd

dataFrame = pd.DataFrame({
    "Car": ['BMW', 'Audi', 'BMW', 'Lexus', 'Tesla', 'Lexus', 'Mustang'],
    "Place": ['Delhi', 'Bangalore', 'Hyderabad', 'Chandigarh', 'Pune', 'Mumbai', 'Jaipur'],
    "Units": [100, 150, 50, 110, 90, 120, 80]
})

print("DataFrame ...\n", dataFrame)
print("\nDisplay all the column names ...\n", dataFrame.columns)

ผลลัพธ์

DataFrame ...
       Car       Place  Units
0      BMW     Delhi    100
1    Audi  Bangalore    150
2      BMW  Hyderabad     50
3   Lexus Chandigarh    110
4   Tesla      Pune     90
5   Lexus    Mumbai    120
6  Mustang    Jaipur     80

Display all the column names ...
Index(['Car', 'Place', 'Units'], dtype='object')

การใช้งาน columns เพิ่มเติม


# แปลงเป็น list
list(df.columns)

# เปลี่ยนชื่อคอลัมน์
df.columns = ['A', 'B', 'C']
# หรือ
df.rename(columns={'Car': 'Vehicle'}, inplace=True)

# เพิ่ม/ลบคอลัมน์
df['NewCol'] = 0
df.drop(columns=['Place'], inplace=True)

# ตรวจสอบคอลัมน์มีอยู่
'Car' in df.columns  # True

ข้อมูลเพิ่มเติม

  • df.columns คืน pd.Index object
  • รองรับ set operations: df1.columns.intersection(df2.columns)
  • MultiIndex columns: df.columns.levels, df.columns.codes