Introdução ao trabalho com arquivos CSV

Arquivos CSV (valores separados por vírgulas) são um formato comum para armazenar dados tabulares. Python fornece o módulo csv para ler, gravar e manipular arquivos CSV com eficiência.

Nesta lição, você aprenderá a trabalhar com arquivos CSV usando o módulo integrado csv do Python.

O que são arquivos CSV?

Arquivos CSV são arquivos de texto que usam um delimitador (geralmente uma vírgula) para separar valores. Cada linha representa uma linha da tabela, e cada valor da linha representa uma coluna.

Lendo arquivos CSV

Para ler um arquivo CSV, use a função csv.reader. Essa função lê cada linha como uma lista de valores:

# 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

Gravando arquivos CSV

Para gravar em um arquivo CSV, use a função csv.writer. Essa função permite gravar linhas em um arquivo CSV:

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

Lendo arquivos CSV com cabeçalhos

Use csv.DictReader para ler arquivos CSV em dicionários, nos quais as chaves são os cabeçalhos das colunas:

# 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

Escrevendo arquivos CSV com cabeçalhos

Use csv.DictWriter para escrever dicionários em arquivos CSV com cabeçalhos:

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

Delimitadores personalizados em arquivos CSV

Você pode especificar um delimitador personalizado ao ler ou gravar arquivos CSV. Por exemplo, arquivos delimitados por tabulação usam uma tabulação (\t) como delimitador:

# 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

Tratamento de arquivos CSV grandes

Para arquivos CSV grandes, use um laço para processar as linhas gradualmente, evitando carregar o arquivo inteiro na memória:

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

Erros comuns com arquivos CSV

Veja alguns erros que devem ser evitados:

• Esquecer de usar newline="" ao gravar arquivos CSV no Windows.
• Usar delimitadores incorretos para arquivos CSV fora do padrão.
• Não tratar dados ausentes ou malformados no arquivo.

O que aprendemos?

Nesta lição, você aprendeu:

• Como ler e gravar arquivos CSV usando o módulo csv do Python.
• Como trabalhar com arquivos CSV que têm cabeçalhos usando DictReader e DictWriter.
• Como tratar arquivos CSV grandes com eficiência.

Muito bem! Vamos passar ao próximo tópico.