Jak wyeksportować tablicę numpy do pliku csv (z przykładami)

Aby wyeksportować tablicę NumPy do pliku CSV, możesz użyć następującej podstawowej składni:

 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=" , ")

Poniższe przykłady pokazują, jak używać tej składni w praktyce.

Przykład 1: Eksportuj tablicę NumPy do formatu CSV

Poniższy kod pokazuje, jak wyeksportować tablicę NumPy do pliku 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=" , ")

Jeśli przejdę do miejsca, w którym na moim laptopie zapisany jest plik CSV, będę mógł wyświetlić dane:

Przykład 2: Eksportuj tablicę NumPy do pliku CSV w określonym formacie

Domyślny format liczb to „%.18e” – wyświetla 18 zer. Możemy jednak użyć argumentu fmt , aby określić inny format.

Na przykład poniższy kod eksportuje tablicę NumPy do formatu CSV i określa dwa miejsca po przecinku:

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

Jeśli przejdę do miejsca, w którym zapisany jest plik CSV, mogę wyświetlić dane:

Przykład 3: Eksportuj tablicę NumPy do pliku CSV z nagłówkami

Poniższy kod pokazuje, jak wyeksportować tablicę NumPy do pliku CSV z niestandardowymi nagłówkami kolumn:

 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="")

Uwaga : Argument komentarzy zapobiega wyświetlaniu symbolu „#” w nagłówkach.

Jeśli przejdę do miejsca, w którym zapisany jest plik CSV, mogę wyświetlić dane:

Uwaga : pełną dokumentację funkcji numpy.savetxt() można znaleźć tutaj .

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje odczytu i zapisu w Pythonie:

Jak czytać pliki CSV za pomocą NumPy
Jak czytać pliki CSV za pomocą Pand
Jak wyeksportować ramkę danych Pandas do pliku CSV