Lendo e escrevendo arquivos CSV em Python
Arquivos CSV (Comma-Separated Values) são amplamente usados para armazenamento e troca de dados. Em Python, você pode facilmente ler e escrever em arquivos CSV usando o módulo csv
integrado. Este artigo o guiará pelos conceitos básicos de trabalho com arquivos CSV em Python.
Lendo arquivos CSV
Para ler um arquivo CSV, você pode usar a classe csv.reader
do módulo csv
. Aqui está um exemplo simples:
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)
Neste exemplo, example.csv
é aberto no modo de leitura, e csv.reader
é usado para analisar o arquivo. Cada linha do arquivo CSV é impressa como uma lista de valores.
Escrevendo arquivos CSV
Para gravar dados em um arquivo CSV, use a classe csv.writer
. Veja como você pode gravar uma lista de linhas em um arquivo 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)
Neste exemplo, output.csv
é criado ou sobrescrito com os dados especificados. O método writer.writerows()
é usado para escrever várias linhas de uma vez.
Conclusão
Trabalhar com arquivos CSV em Python é direto usando o módulo csv
. Você pode facilmente ler e escrever em arquivos CSV, tornando simples lidar com dados neste formato comum.