Python 的 CSV 模块指南

Die Arbeit mit Daten ist ein unvermeidlicher Teil der Programmierung, und als jemand, der oft knietief in verschiedenen Dateiformaten steckt, habe ich immer geschätzt, wie Python den gesamten Prozess vereinfacht.

Ein solches Dateiformat, das insbesondere in der Datenanalyse regelmäßig auftaucht, ist die CSV-Datei.

CSV (Comma-Separated Values) ist aufgrund seiner Einfachheit ein beliebtes Datenaustauschformat.

Glücklicherweise verfügt Python über ein integriertes Modul namens csv, das die Arbeit mit diesen Dateien äußerst effizient macht.

In diesem Artikel erläutere ich die Funktionsweise des CSV-Moduls in Python, von der grundlegenden Verwendung bis hin zu fortgeschritteneren Techniken, mit denen Sie bei der Datenverarbeitung jede Menge Zeit sparen können.

---

Was ist eine CSV-Datei?

Bevor wir uns mit dem CSV-Modul befassen, beginnen wir mit einem grundlegenden Verständnis dessen, was eine CSV-Datei ist.

Eine CSV-Datei ist im Wesentlichen eine reine Textdatei, in der jede Zeile eine Datenzeile darstellt und jeder Wert durch ein Komma (oder manchmal andere Trennzeichen wie Tabulatoren) getrennt ist.

Hier ist ein kurzes Beispiel, wie es aussehen könnte:

Name,Age,Occupation
Alice,30,Engineer
Bob,25,Data Scientist
Charlie,35,Teacher

---

Warum das CSV-Modul?

Sie fragen sich vielleicht, warum Sie das CSV-Modul benötigen, wenn CSV-Dateien nur Textdateien sind, die theoretisch mit Pythons Standardmethoden zur Dateiverarbeitung gelesen werden könnten.

Obwohl dies zutrifft, können CSV-Dateien Komplexitäten aufweisen – wie eingebettete Kommas, Zeilenumbrüche innerhalb von Zellen und unterschiedliche Trennzeichen –, die manuell nur schwer zu handhaben sind.

Das CSV-Modul abstrahiert all dies, sodass Sie sich auf Ihre Daten konzentrieren können.

---

CSV-Dateien lesen

Lassen Sie uns zum Code springen.

Der häufigste Vorgang, den Sie an einer CSV-Datei ausführen, ist das Lesen ihres Inhalts.

Die Funktion csv.reader() im Modul ist hierfür ein einfach zu verwendendes Tool.

Hier finden Sie eine Schritt-für-Schritt-Anleitung, wie es geht.

Grundlegende CSV-Lesung

import csv

# Open a CSV file
with open('example.csv', 'r') as file:
    reader = csv.reader(file)

    # Iterate over the rows
    for row in reader:
        print(row)

Dies ist die einfachste Möglichkeit, eine CSV-Datei zu lesen.

Der csv.reader() gibt eine Iterable zurück, wobei jede Iteration Ihnen eine Liste liefert, die eine Zeile der Datei darstellt.

Verwaltung von Headern
Die meisten CSV-Dateien enthalten in der ersten Zeile Kopfzeilen, ähnlich wie Spaltennamen.

Wenn Sie diese Überschriften nicht benötigen, können Sie beim Iterieren einfach die erste Zeile überspringen:

import csv

with open('example.csv', 'r') as file:
    reader = csv.reader(file)

    # Skip header
    next(reader)

    for row in reader:
        print(row)

Manchmal arbeite ich mit Dateien, die eine Mischung aus nützlichen und irrelevanten Daten enthalten, und überspringe Zeilen, die nicht nur auf der Kopfzeile basieren.

Sie können dies ganz einfach innerhalb der for-Schleife tun.

DictReader: Eine intuitivere Möglichkeit, CSV-Dateien zu lesen
Wenn Ihre CSV-Datei Kopfzeilen hat, ist csv.DictReader() eine weitere fantastische Option, die jede Zeile als Wörterbuch liest, wobei die Schlüssel die Spaltennamen sind:

import csv

with open('example.csv', 'r') as file:
    reader = csv.DictReader(file)

    for row in reader:
        print(row)

Dieser Ansatz kann Ihren Code lesbarer und intuitiver machen, insbesondere wenn Sie mit großen Datensätzen arbeiten.

Zum Beispiel fühlt sich der Zugriff auf Zeile['Name'] viel klarer an als der Umgang mit indexbasierten Zugriffen wie Zeile[0].

---

Schreiben in CSV-Dateien

Sobald Sie Ihre Daten gelesen und verarbeitet haben, möchten Sie sie wahrscheinlich speichern oder exportieren.

Die Funktion csv.writer() ist Ihr bevorzugtes Werkzeug zum Schreiben in CSV-Dateien.

Grundlegendes CSV-Schreiben

import csv

# Data to be written
data = [
    ['Name', 'Age', 'Occupation'],
    ['Alice', 30, 'Engineer'],
    ['Bob', 25, 'Data Scientist'],
    ['Charlie', 35, 'Teacher']
]

# Open a file in write mode
with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)

    # Write data to the file
    writer.writerows(data)

Die Funktion write.writerows() nimmt eine Liste von Listen und schreibt sie in die CSV-Datei, wobei jede innere Liste eine Datenzeile darstellt.

DictWriter: Eine sauberere Möglichkeit, CSV-Dateien zu schreiben
So wie wir DictReader zum Einlesen von CSV-Dateien in Wörterbücher haben, haben wir DictWriter zum Schreiben von Wörterbüchern in eine CSV-Datei.

Diese Methode kann besonders praktisch sein, wenn Sie Ihre Spaltennamen explizit angeben möchten.

import csv

# Data as list of dictionaries
data = [
    {'Name': 'Alice', 'Age': 30, 'Occupation': 'Engineer'},
    {'Name': 'Bob', 'Age': 25, 'Occupation': 'Data Scientist'},
    {'Name': 'Charlie', 'Age': 35, 'Occupation': 'Teacher'}
]

# Open file for writing
with open('output.csv', 'w', newline='') as file:
    fieldnames = ['Name', 'Age', 'Occupation']
    writer = csv.DictWriter(file, fieldnames=fieldnames)

    # Write the header
    writer.writeheader()

    # Write the data
    writer.writerows(data)

Mit DictWriter erhalten Sie eine schöne, übersichtliche Oberfläche zum Schreiben von Wörterbüchern in CSV, während Ihr Code lesbar und prägnant bleibt.

---

Trennzeichen anpassen

Standardmäßig verwendet das CSV-Modul Kommas, um Werte zu trennen, aber manchmal arbeiten Sie möglicherweise mit Dateien, die andere Trennzeichen wie Tabulatoren oder Semikolons verwenden.

Das CSV-Modul bietet eine einfache Möglichkeit, diese Fälle zu handhaben, indem das Trennzeichenargument angegeben wird.

import csv

with open('example_tab.csv', 'r') as file:
    reader = csv.reader(file, delimiter='\t')

    for row in reader:
        print(row)

Ich bin auf CSV-Dateien gestoßen, die Semikolons anstelle von Kommas verwenden – normalerweise aus europäischen Quellen – und es ist beruhigend zu wissen, dass das CSV-Modul von Python dies problemlos bewältigt.

Ob Kommas, Tabulatoren oder andere Trennzeichen, mit dem CSV-Modul sind Sie bestens gerüstet.

---

Umgang mit komplexen Daten

Was passiert, wenn Ihre Daten Kommas in Feldern, Anführungszeichen oder sogar Zeilenumbrüche enthalten?

Das CSV-Modul behandelt solche Fälle automatisch mithilfe von Anführungszeichenmechanismen.

Sie können die Funktionsweise der Angebotserstellung auch mithilfe des Angebotsparameters steuern.

import csv

data = [
    ['Name', 'Occupation', 'Description'],
    ['Alice', 'Engineer', 'Works on, "cutting-edge" technology'],
    ['Bob', 'Data Scientist', 'Loves analyzing data.']
]

with open('complex.csv', 'w', newline='') as file:
    writer = csv.writer(file, quoting=csv.QUOTE_ALL)
    writer.writerows(data)

In this example, QUOTE_ALL ensures that every field is wrapped in quotes.

Other quoting options include csv.QUOTE_MINIMAL, csv.QUOTE_NONNUMERIC, and csv.QUOTE_NONE, giving you full control over how your CSV data is formatted.

---

Conclusion

Over the years, I’ve come to rely on the CSV format as a lightweight, efficient way to move data around, and Python’s csv module has been a trusty companion in that journey.

Whether you’re dealing with simple spreadsheets or complex, multi-line data fields, this module makes the process feel intuitive and effortless.

While working with CSVs may seem like a mundane task at first, it’s a gateway to mastering data manipulation.

In my experience, once you’ve conquered CSVs, you'll find yourself confidently tackling larger, more complex formats like JSON or SQL databases. After all, everything starts with the basics.

以上是Python 的 CSV 模块指南的详细内容。更多信息请关注PHP中文网其他相关文章！