Gravando CSV

Gravar CSV do jeito básico

Gravar CSV segue a mesma lógica de ler, ao contrário. Você abre o arquivo em modo de escrita, com o 'w' como segundo argumento, cria um escritor em cima dele e chama métodos para despejar as linhas. O modo 'w' cria o arquivo se ele não existe e apaga o conteúdo se ele já existe, então tome cuidado para não sobrescrever algo importante. O newline='' continua obrigatório, pela mesma razão da leitura. Veja a forma mais direta, com csv.writer, que recebe cada linha como uma lista.

import csv

linhas = [
    ["nome", "nota", "cidade"],
    ["Ana", 9.5, "Recife"],
    ["Bruno", 7.0, "Salvador"],
]

with open("saida.csv", "w", newline="", encoding="utf-8-sig") as arquivo:
    escritor = csv.writer(arquivo)
    for linha in linhas:
        escritor.writerow(linha)

print("Arquivo gravado.")  # Arquivo gravado.

csv.writer grava listas; cada writerow escreve uma linha do arquivo.

O método writerow escreve uma linha por vez, cuidando de juntar os campos com vírgula e de colocar aspas quando um valor tem vírgula ou quebra de linha dentro. Note que aqui os números não precisam virar texto na mão: o writer converte 9.5 para o texto certo no arquivo. Se você tem uma lista de listas pronta, dá para gravar todas de uma vez com writerows, sem o laço, como você vê mais adiante. O resultado é um CSV que qualquer planilha abre.

DictWriter e o cabeçalho automático

Quando seus dados já estão em dicionários, o que é comum depois de ler com DictReader ou de montar registros no programa, o csv.DictWriter é o par ideal. Você informa os nomes das colunas em fieldnames, chama writeheader uma vez para escrever a primeira linha com esses nomes, e depois passa cada dicionário para writerow. O DictWriter garante que cada campo caia na coluna certa pelo nome, mesmo que os dicionários estejam em ordem diferente.

import csv

registros = [
    {"nome": "Ana", "nota": 9.5, "cidade": "Recife"},
    {"nome": "Bruno", "nota": 7.0, "cidade": "Salvador"},
]

colunas = ["nome", "nota", "cidade"]

with open("saida.csv", "w", newline="", encoding="utf-8-sig") as arquivo:
    escritor = csv.DictWriter(arquivo, fieldnames=colunas)
    escritor.writeheader()
    escritor.writerows(registros)

print("Gravado com cabecalho.")  # Gravado com cabecalho.

DictWriter: fieldnames define as colunas, writeheader escreve o cabeçalho, writerows grava tudo.

Duas coisas se destacam. O writeheader escreve automaticamente nome,nota,cidade como primeira linha, usando exatamente os fieldnames que você definiu, sem precisar digitar o cabeçalho de novo. E o writerows recebe a lista inteira de dicionários e grava todos de uma vez, mais enxuto que um laço com writerow. Se um dicionário tiver uma chave a mais que não está em fieldnames, o DictWriter avisa com erro, o que ajuda a pegar campos digitados errado. Esse é o padrão que fecha o relatório do fim do módulo.

Acentos e o Excel: por que utf-8-sig

Aqui mora uma frustração clássica de quem grava CSV no Brasil. Você salva um arquivo com utf-8, abre no Excel em português e os acentos viram símbolos estranhos: São Paulo aparece como algo ilegível. O problema não é o Python, é o Excel: ele, por padrão, não assume utf-8 ao abrir CSV. A solução é gravar com encoding='utf-8-sig'. O sufixo sig adiciona uma marca invisível no início do arquivo, o BOM, que o Excel reconhece e então mostra os acentos corretamente.

A regra prática é simples. Se o CSV que você grava vai ser aberto por uma pessoa no Excel em português, use utf-8-sig e evite a dor de cabeça dos acentos. Se ele vai ser lido por outro programa, ou pelo próprio Python, o utf-8 comum basta e é o mais limpo. Nas duas situações, na leitura você continua usando utf-8, que entende o arquivo com ou sem a marca. É um detalhe pequeno, mas é a diferença entre um relatório que abre bonito e um que abre bagunçado.