Odczytywanie i zapisywanie plików CSV w Pythonie

Pliki CSV (Comma-Separated Values) są szeroko stosowane do przechowywania i wymiany danych. W Pythonie możesz łatwo czytać i zapisywać pliki CSV, korzystając z wbudowanego modułu csv. Ten artykuł przeprowadzi Cię przez podstawy pracy z plikami CSV w Pythonie.

Odczytywanie plików CSV

Aby odczytać plik CSV, możesz użyć klasy csv.reader z modułu csv. Oto prosty przykład:

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)

W tym przykładzie example.csv jest otwierany w trybie odczytu, a csv.reader jest używany do parsowania pliku. Każdy wiersz pliku CSV jest drukowany jako lista wartości.

Pisanie plików CSV

Aby zapisać dane w pliku CSV, użyj klasy csv.writer. Oto jak możesz zapisać listę wierszy w pliku 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)

W tym przykładzie plik output.csv jest tworzony lub nadpisywany określonymi danymi. Metoda writer.writerows() jest używana do zapisywania wielu wierszy jednocześnie.

Wniosek

Praca z plikami CSV w Pythonie jest prosta dzięki modułowi csv. Możesz łatwo czytać i zapisywać pliki CSV, co ułatwia obsługę danych w tym powszechnym formacie.