Đọc và ghi tệp CSV trong Python

Tệp CSV (Giá trị phân cách bằng dấu phẩy) được sử dụng rộng rãi để lưu trữ và trao đổi dữ liệu. Trong Python, bạn có thể dễ dàng đọc và ghi vào tệp CSV bằng mô-đun csv tích hợp sẵn. Bài viết này sẽ hướng dẫn bạn những điều cơ bản khi làm việc với tệp CSV trong Python.

Đọc tập tin CSV

Để đọc tệp CSV, bạn có thể sử dụng lớp csv.reader từ mô-đun csv. Sau đây là một ví dụ đơn giản:

import csv

# Open the CSV file
with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    
    # Iterate over each row in the CSV
    for row in reader:
        print(row)

Trong ví dụ này, example.csv được mở ở chế độ đọc và csv.reader được sử dụng để phân tích tệp. Mỗi hàng của tệp CSV được in dưới dạng danh sách các giá trị.

Viết tệp CSV

Để ghi dữ liệu vào tệp CSV, hãy sử dụng lớp csv.writer. Sau đây là cách bạn có thể ghi danh sách các hàng vào tệp CSV:

import csv

# Data to be written to the CSV
data = [
    ['Name', 'Age', 'City'],
    ['Alice', 30, 'New York'],
    ['Bob', 25, 'Los Angeles'],
    ['Charlie', 35, 'Chicago']
]

# Open the CSV file in write mode
with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    
    # Write each row to the CSV file
    writer.writerows(data)

Trong ví dụ này, output.csv được tạo hoặc ghi đè bằng dữ liệu đã chỉ định. Phương thức writer.writerows() được sử dụng để ghi nhiều hàng cùng một lúc.

Phần kết luận

Làm việc với các tệp CSV trong Python rất đơn giản khi sử dụng mô-đun csv. Bạn có thể dễ dàng đọc và ghi vào các tệp CSV, giúp xử lý dữ liệu ở định dạng phổ biến này một cách đơn giản.