วิธีรับชื่อคอลัมน์ใน pandas (3 วิธี)

คุณสามารถใช้วิธีการต่อไปนี้เพื่อรับชื่อคอลัมน์ใน DataFrame แพนด้า:

วิธีที่ 1: รับชื่อคอลัมน์ทั้งหมด

 list (df)

วิธีที่ 2: รับชื่อคอลัมน์ตามตัวอักษร

 sorted (df)

วิธีที่ 3: รับชื่อคอลัมน์ที่มีชนิดข้อมูลเฉพาะ

 list ( df.select_dtypes (include=[' int64 ', ' bool ']))

ตัวอย่างต่อไปนี้แสดงวิธีการใช้แต่ละวิธีกับ DataFrame แพนด้าต่อไปนี้:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F'],
                   ' points ': [18, 22, 19, 14, 14, 11],
                   ' assists ': [5, 7, 7, 9, 12, 9],
                   ' playoffs ': [True, False, False, True, True, True]})

#view DataFrame
print (df)

  team points assists playoffs
0 A 18 5 True
1 B 22 7 False
2 C 19 7 False
3 D 14 9 True
4 E 14 12 True
5 F 11 9 True

ตัวอย่างที่ 1: รับชื่อคอลัมน์ทั้งหมด

วิธีที่ง่ายที่สุดในการรับชื่อคอลัมน์ทั้งหมดใน DataFrame ของ pandas คือการใช้ list() ดังนี้:

 #get all column names
list (df)

['team', 'points', 'assists', 'playoffs']

ผลลัพธ์คือรายการที่มีชื่อสี่คอลัมน์ของ pandas DataFrame

ตัวอย่างที่ 2: รับชื่อคอลัมน์ตามตัวอักษร

หากต้องการรับชื่อคอลัมน์ใน DataFrame ของ pandas ตามลำดับตัวอักษร คุณสามารถใช้ฟังก์ชัน sorted() ได้ดังนี้:

 #get column names in alphabetical order
sorted (df)

['assists', 'playoffs', 'points', 'team']

ผลลัพธ์คือรายการที่มีชื่อสี่คอลัมน์ของ Pandas DataFrame ที่แสดงตามลำดับตัวอักษร

คุณยังสามารถใช้อาร์กิวเมนต์ Reverse=True เพื่อรับชื่อคอลัมน์ตามลำดับตัวอักษรย้อนกลับได้:

 #get column names in reverse alphabetical order
sorted (df, reverse= True )

['team', 'points', 'playoffs', 'assists']

ตัวอย่างที่ 3: รับชื่อคอลัมน์ที่มีชนิดข้อมูลเฉพาะ

คุณสามารถใช้ไวยากรณ์ต่อไปนี้เพื่อแสดงชนิดข้อมูลของแต่ละคอลัมน์ใน DataFrame:

 #view data type of each column
df. dtypes

team object
int64 dots
assists int64
playoffs bool
dtype:object

จากนั้นคุณสามารถใช้ฟังก์ชัน select_dtypes() เพื่อรับเฉพาะชื่อคอลัมน์ที่มีประเภทข้อมูลเฉพาะได้

ตัวอย่างเช่น เราสามารถใช้ไวยากรณ์ต่อไปนี้เพื่อรับเฉพาะชื่อคอลัมน์ที่มีชนิดข้อมูล int64 หรือ bool :

 #get all columns that have data type of int64 or bool
list ( df.select_dtypes (include=[' int64 ', ' bool ']))

['points', 'assists', 'playoffs']

ผลลัพธ์คือรายการชื่อคอลัมน์ที่มีประเภทข้อมูล int64 หรือ บูล

แหล่งข้อมูลเพิ่มเติม

บทช่วยสอนต่อไปนี้จะอธิบายวิธีดำเนินการทั่วไปอื่น ๆ ในแพนด้า:

Pandas: วิธีเปลี่ยนชื่อคอลัมน์
Pandas: วิธีตั้งค่าคอลัมน์เป็นดัชนี
Pandas: รับดัชนีของแถวที่มีคอลัมน์ตรงกับค่า