Läsa och skriva CSV-filer i Python
CSV-filer (Comma-Separated Values) används ofta för datalagring och utbyte. I Python kan du enkelt läsa från och skriva till CSV-filer med den inbyggda csv-modulen. Den här artikeln guidar dig genom grunderna för att arbeta med CSV-filer i Python.
Läser CSV-filer
För att läsa en CSV-fil kan du använda klassen csv.reader från modulen csv. Här är ett enkelt exempel:
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)I det här exemplet öppnas example.csv i läsläge och csv.reader används för att analysera filen. Varje rad i CSV-filen skrivs ut som en lista med värden.
Skriva CSV-filer
För att skriva data till en CSV-fil, använd klassen csv.writer. Så här kan du skriva en lista med rader till en CSV-fil:
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)I det här exemplet skapas eller skrivs output.csv över med angivna data. Metoden writer.writerows() används för att skriva flera rader samtidigt.
Slutsats
Att arbeta med CSV-filer i Python är enkelt med csv-modulen. Du kan enkelt läsa från och skriva till CSV-filer, vilket gör det enkelt att hantera data i detta vanliga format.