Arbeta med CSV-filer

CSV-filer (kommaseparerade värden) är ett vanligt format för att lagra tabelldata. Python tillhandahåller modulen csv för att effektivt läsa, skriva och bearbeta CSV-filer.

I den här lektionen lär du dig att hantera CSV-filer med hjälp av Pythons inbyggda modul csv.

CSV-filer är textfiler som använder en avgränsare (vanligtvis ett komma) för att separera värden. Varje rad motsvarar en rad i tabellen och varje värde på raden motsvarar en kolumn.

För att läsa en CSV-fil använder du funktionen csv.reader. Funktionen läser varje rad som en lista med värden:

# 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

För att skriva till en CSV-fil använder du funktionen csv.writer. Funktionen gör det möjligt att skriva rader till en CSV-fil:

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

Använd csv.DictReader för att läsa in CSV-filer i ordböcker, där nycklarna är kolumnrubrikerna:

# 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

Använd csv.DictWriter för att skriva ordböcker till CSV-filer med rubriker:

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

Du kan ange en anpassad avgränsare när du läser eller skriver CSV-filer. Tabavgränsade filer använder till exempel en tabb (\t) som avgränsare:

# 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

För stora CSV-filer bör du använda en loop för att bearbeta rader stegvis och undvika att läsa in hela filen i minnet:

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

Här är några misstag som bör undvikas:

• Att glömma att använda newline="" när du skriver CSV-filer i Windows.
• Att använda felaktiga avgränsare för CSV-filer som inte följer standardformatet.
• Att inte hantera data som saknas eller är felaktigt formaterade i filen.

I den här lektionen har du lärt dig:

• Hur du läser och skriver CSV-filer med Pythons modul csv.
• Hur du arbetar med CSV-filer som har rubriker med hjälp av DictReader och DictWriter.
• Hur du hanterar stora CSV-filer effektivt.

Bra jobbat! Nu går vi vidare till nästa ämne.