Pandas: วิธีระบุประเภทเมื่อนำเข้าไฟล์ csv

คุณสามารถใช้ไวยากรณ์พื้นฐานต่อไปนี้เพื่อระบุประเภทของแต่ละคอลัมน์ใน DataFrame เมื่อนำเข้าไฟล์ CSV ลงในแพนด้า:

 df = pd. read_csv (' my_data.csv ',
                 dtype = {' col1 ': str , ' col2 ': float , ' col3 ': int })

อาร์กิวเมนต์ dtype ระบุประเภทข้อมูลที่แต่ละคอลัมน์ควรมีเมื่อนำเข้าไฟล์ CSV ไปยัง DataFrame ของแพนด้า

ตัวอย่างต่อไปนี้แสดงวิธีใช้ไวยากรณ์นี้ในทางปฏิบัติ

ตัวอย่าง: ระบุประเภทเมื่อนำเข้าไฟล์ CSV ไปยัง Pandas

สมมติว่าเรามีไฟล์ CSV ต่อไปนี้ชื่อ Basketball_data.csv :

หากเรานำเข้าไฟล์ CSV โดยใช้ฟังก์ชัน read_csv() แพนด้าจะพยายามระบุประเภทข้อมูลสำหรับแต่ละคอลัมน์โดยอัตโนมัติ:

 import pandas as pd

#import CSV file
df = pd. read_csv (' basketball_data.csv ')

#view resulting DataFrame
print (df)

   At 22 10
0 B 14 9
1 C 29 6
2 D 30 2
3 E 22 9
4 F 31 10

#view data type of each column
print ( df.dtypes )

team object
int64 dots
rebounds int64
dtype:object

จากผลลัพธ์เราจะเห็นว่าคอลัมน์ของ DataFrame มีประเภทข้อมูลดังต่อไปนี้

• ทีม : object
• คะแนน : int64
• ตีกลับ : int64

อย่างไรก็ตาม เราสามารถใช้อาร์กิวเมนต์ dtype ในฟังก์ชัน read_csv() เพื่อระบุประเภทข้อมูลที่แต่ละคอลัมน์ควรมี:

import pandas as pd

#import CSV file and specify dtype of each column
df = pd. read_csv (' basketball_data.csv ',
                 dtype = {' team ': str , ' points ': float , ' rebounds ': int }))

#view resulting DataFrame
print (df)

   At 22 10
0 B 14 9
1 C 29 6
2 D 30 2
3 E 22 9
4 F 31 10

#view data type of each column
print ( df.dtypes )

team object
float64 points
rebounds int32
dtype:object

จากผลลัพธ์เราจะเห็นว่าคอลัมน์ของ DataFrame มีประเภทข้อมูลดังต่อไปนี้

• ทีม : object
• คะแนน : float64
• การตีกลับ : int32

ประเภทข้อมูลเหล่านี้สอดคล้องกับประเภทข้อมูลที่เราระบุโดยใช้อาร์กิวเมนต์ dtype

โปรดทราบว่าในตัวอย่างนี้ เราได้ระบุประเภทสำหรับแต่ละคอลัมน์ใน DataFrame

อย่างไรก็ตาม คุณสามารถเลือกระบุประเภทสำหรับคอลัมน์ที่ต้องการเท่านั้น และให้แพนด้าอนุมานประเภทของคอลัมน์ที่เหลือได้

หมายเหตุ : คุณสามารถดูเอกสารฉบับเต็มของฟังก์ชัน pandas read_csv() ได้ที่นี่

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

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

Pandas: วิธีข้ามบรรทัดเมื่ออ่านไฟล์ CSV
Pandas: วิธีเพิ่มข้อมูลลงในไฟล์ CSV ที่มีอยู่
Pandas: วิธีอ่านไฟล์ CSV ที่ไม่มีส่วนหัว
Pandas: วิธีตั้งชื่อคอลัมน์เมื่อนำเข้าไฟล์ CSV