Python:
Trabalhando com CSV
Como fazer:
Python fornece o módulo integrado csv
para manipular arquivos CSV, tornando simples ler e escrever neles. Para manipulação de dados mais robusta e complexa, a biblioteca de terceiros pandas
é muito popular.
Usando o módulo csv
Lendo um arquivo CSV
import csv
with open('sample.csv', mode='r') as file:
leitor_csv = csv.reader(file)
for row in leitor_csv:
print(row)
Assumindo que sample.csv
contém:
name,age,city
John,22,New York
Jane,28,Los Angeles
Saída:
['name', 'age', 'city']
['John', '22', 'New York']
['Jane', '28', 'Los Angeles']
Escrevendo em um arquivo CSV
import csv
linhas = [['name', 'age', 'city'], ['Jack', '33', 'Chicago'], ['Emily', '41', 'Denver']]
with open('output.csv', mode='w', newline='') as file:
escritor = csv.writer(file)
escritor.writerows(linhas)
Cria ou sobrescreve output.csv
com:
name,age,city
Jack,33,Chicago
Emily,41,Denver
Usando pandas
para CSV
pandas
é uma biblioteca poderosa para manipulação de dados que simplifica trabalhar com arquivos CSV entre outros formatos de dados.
Instalar pandas
pip install pandas
Lendo um arquivo CSV com pandas
import pandas as pd
df = pd.read_csv('sample.csv')
print(df)
Saída:
name age city
0 John 22 New York
1 Jane 28 Los Angeles
Escrevendo em um arquivo CSV com pandas
import pandas as pd
df = pd.DataFrame({'name': ['Jack', 'Emily'], 'age': [33, 41], 'city': ['Chicago', 'Denver']})
df.to_csv('output_pandas.csv', index=False)
Cria ou sobrescreve output_pandas.csv
com:
name,age,city
Jack,33,Chicago
Emily,41,Denver