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.