บทนำสู่การทำงานกับไฟล์ CSV

ไฟล์ CSV (ค่าที่คั่นด้วยจุลภาค) เป็นรูปแบบทั่วไปสำหรับจัดเก็บข้อมูลแบบตาราง Python มีมอดูล csv สำหรับอ่าน เขียน และจัดการไฟล์ CSV ได้อย่างมีประสิทธิภาพ

ในบทเรียนนี้ คุณจะได้เรียนรู้วิธีจัดการไฟล์ CSV โดยใช้มอดูล csv ในตัวของ Python

ไฟล์ CSV คืออะไร

ไฟล์ CSV เป็นไฟล์ข้อความที่ใช้ตัวคั่น (โดยปกติคือจุลภาค) เพื่อแยกค่า แต่ละบรรทัดจะแทนหนึ่งแถว และแต่ละค่าในบรรทัดจะแทนหนึ่งคอลัมน์

การอ่านไฟล์ CSV

หากต้องการอ่านไฟล์ CSV ให้ใช้ฟังก์ชัน csv.reader ฟังก์ชันนี้จะอ่านแต่ละบรรทัดเป็นรายการค่า:

# Reading a CSV file
import csv

with open("data.csv", "r") as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)  # Each row is a list of values

การเขียนไฟล์ CSV

หากต้องการเขียนลงในไฟล์ CSV ให้ใช้ฟังก์ชัน csv.writer ฟังก์ชันนี้ช่วยให้คุณเขียนแถวลงในไฟล์ CSV ได้:

# Writing to a CSV file
import csv

with open("output.csv", "w", newline="") as file:
    writer = csv.writer(file)
    writer.writerow(["Name", "Age", "City"])
    writer.writerow(["Alice", 30, "New York"])
    writer.writerow(["Bob", 25, "London"])

การอ่านไฟล์ CSV ที่มีส่วนหัว

ใช้ csv.DictReader เพื่ออ่านไฟล์ CSV เป็นพจนานุกรม โดยมีชื่อส่วนหัวคอลัมน์เป็นคีย์:

# Reading a CSV file with headers
import csv

with open("data.csv", "r") as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row)  # Each row is a dictionary with column headers as keys

การเขียนไฟล์ CSV ที่มีส่วนหัว

ใช้ csv.DictWriter เพื่อเขียนพจนานุกรมลงในไฟล์ CSV ที่มีส่วนหัว:

# Writing a CSV file with headers
import csv

with open("output.csv", "w", newline="") as file:
    fieldnames = ["Name", "Age", "City"]
    writer = csv.DictWriter(file, fieldnames=fieldnames)

    writer.writeheader()  # Write the header row
    writer.writerow({"Name": "Alice", "Age": 30, "City": "New York"})
    writer.writerow({"Name": "Bob", "Age": 25, "City": "London"})

ตัวคั่นแบบกำหนดเองในไฟล์ CSV

คุณสามารถระบุตัวคั่นแบบกำหนดเองเมื่ออ่านหรือเขียนไฟล์ CSV ได้ ตัวอย่างเช่น ไฟล์ที่คั่นด้วยแท็บจะใช้แท็บ (\t) เป็นตัวคั่น:

# Using a custom delimiter
import csv

with open("data.tsv", "r") as file:
    reader = csv.reader(file, delimiter="\t")
    for row in reader:
        print(row)  # Each row is split by tab

การจัดการไฟล์ CSV ขนาดใหญ่

สำหรับไฟล์ CSV ขนาดใหญ่ ให้ใช้ลูปประมวลผลแถวทีละส่วน เพื่อหลีกเลี่ยงการโหลดไฟล์ทั้งหมดเข้าสู่หน่วยความจำ:

# Reading a large CSV file efficiently
import csv

with open("large_data.csv", "r") as file:
    reader = csv.reader(file)
    for row in reader:
        # Process each row
        print(row)

ข้อผิดพลาดที่พบบ่อยเกี่ยวกับไฟล์ CSV

ต่อไปนี้คือข้อผิดพลาดบางประการที่ควรหลีกเลี่ยง:

• ลืมใช้ newline="" เมื่อเขียนไฟล์ CSV บน Windows
• ใช้ตัวคั่นไม่ถูกต้องสำหรับไฟล์ CSV ที่มีรูปแบบแตกต่างจากมาตรฐาน
• ไม่จัดการข้อมูลที่หายไปหรือมีรูปแบบไม่ถูกต้องในไฟล์

เรียนรู้อะไรบ้าง

ในบทเรียนนี้ คุณได้เรียนรู้ว่า:

• อ่านและเขียนไฟล์ CSV โดยใช้มอดูล csv ของ Python ได้อย่างไร
• ทำงานกับไฟล์ CSV ที่มีส่วนหัวโดยใช้ DictReader และ DictWriter ได้อย่างไร
• จัดการไฟล์ CSV ขนาดใหญ่ได้อย่างมีประสิทธิภาพได้อย่างไร

ทำได้ดีมาก! ไปยังหัวข้อถัดไปกันเถอะ