Como exportar um array numpy para um arquivo csv (com exemplos)

Você pode usar a seguinte sintaxe básica para exportar um array NumPy para um arquivo CSV:

 import numpy as np

#define NumPy array
data = np. array ([[1,2,3],[4,5,6],[7,8,9]])

#export array to CSV file
n.p. savetxt (" my_data.csv ", data, delimiter=" , ")

Os exemplos a seguir mostram como usar essa sintaxe na prática.

Exemplo 1: Exportar um array NumPy para o formato CSV

O código a seguir mostra como exportar um array NumPy para um arquivo CSV:

 import numpy as np

#define NumPy array
data = np. array ([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])

#export array to CSV file
n.p. savetxt (" my_data.csv ", data, delimiter=" , ")

Se eu navegar até onde o arquivo CSV está salvo em meu laptop, posso visualizar os dados:

Exemplo 2: exporte um array NumPy para CSV com um formato específico

O formato padrão para números é “%.18e” – exibe 18 zeros. Entretanto, podemos usar o argumento fmt para especificar um formato diferente.

Por exemplo, o código a seguir exporta uma matriz NumPy para o formato CSV e especifica duas casas decimais:

 import numpy as np

#define NumPy array
data = np. array ([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])

#export array to CSV file (using 2 decimal places)
n.p. savetxt (" my_data.csv ", data, delimiter=" , ", fmt=" %.2f ")

Se eu navegar até onde o arquivo CSV está salvo, posso visualizar os dados:

Exemplo 3: Exportar um array NumPy para CSV com cabeçalhos

O código a seguir mostra como exportar um array NumPy para um arquivo CSV com cabeçalhos de coluna personalizados:

 import numpy as np

#define NumPy array
data = np. array ([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12], [13, 14, 15]])

#export array to CSV file (using 2 decimal places)
n.p. savetxt (" my_data.csv ", data, delimiter=" , ", fmt=" %.2f ",
           header=" A, B, C ", comments="")

Nota : O argumento de comentários evita que um símbolo “#” seja exibido nos cabeçalhos.

Se eu navegar até onde o arquivo CSV está salvo, posso visualizar os dados:

Nota : Você pode encontrar a documentação completa para a função numpy.savetxt() aqui .

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns de leitura e gravação em Python:

Como ler arquivos CSV com NumPy
Como ler arquivos CSV com Pandas
Como exportar o Pandas DataFrame para um arquivo CSV