Wprowadzenie do pracy z plikami CSV

Pliki CSV (wartości rozdzielane przecinkami) to popularny format przechowywania danych tabelarycznych. Python udostępnia moduł csv umożliwiający wydajne odczytywanie, zapisywanie i przetwarzanie plików CSV.

W tej lekcji dowiedzą się Państwo, jak obsługiwać pliki CSV za pomocą wbudowanego modułu csv języka Python.

Czym są pliki CSV?

Pliki CSV to pliki tekstowe, które używają separatora (zwykle przecinka) do rozdzielania wartości. Każdy wiersz reprezentuje wiersz tabeli, a każda wartość w wierszu reprezentuje kolumnę.

Odczytywanie plików CSV

Aby odczytać plik CSV, należy użyć funkcji csv.reader. Funkcja ta odczytuje każdy wiersz jako listę wartości:

# 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

Zapisywanie plików CSV

Aby zapisać dane do pliku CSV, należy użyć funkcji csv.writer. Funkcja ta umożliwia zapisywanie wierszy do pliku 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"])

Odczytywanie plików CSV z nagłówkami

Użyj csv.DictReader, aby odczytać pliki CSV do słowników, w których kluczami są nagłówki kolumn:

# 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

Zapisywanie plików CSV z nagłówkami

Użyj csv.DictWriter, aby zapisywać słowniki do plików CSV z nagłówkami:

# 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"})

Niestandardowe separatory w plikach CSV

Podczas odczytywania lub zapisywania plików CSV można określić niestandardowy separator. Na przykład w plikach rozdzielanych tabulatorami jako separatora używa się tabulatora (\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

Obsługa dużych plików CSV

W przypadku dużych plików CSV należy użyć pętli do stopniowego przetwarzania wierszy, unikając wczytywania całego pliku do pamięci:

# 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)

Typowe błędy związane z plikami CSV

Oto błędy, których należy unikać:

• Zapominanie o użyciu newline="" podczas zapisywania plików CSV w systemie Windows.
• Używanie nieprawidłowych separatorów w niestandardowych plikach CSV.
• Brak obsługi brakujących lub nieprawidłowo sformatowanych danych w pliku.

Czego się nauczyliśmy?

W tej lekcji dowiedzieli się Państwo:

• Jak odczytywać i zapisywać pliki CSV za pomocą modułu csv języka Python.
• Jak pracować z plikami CSV zawierającymi nagłówki za pomocą DictReader i DictWriter.
• Jak wydajnie obsługiwać duże pliki CSV.

Świetnie! Przejdźmy do następnego tematu.