Numpy 배열을 csv 파일로 내보내는 방법(예제 포함)

다음 기본 구문을 사용하여 NumPy 배열을 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=" , ")

다음 예에서는 이 구문을 실제로 사용하는 방법을 보여줍니다.

예 1: NumPy 배열을 CSV 형식으로 내보내기

다음 코드는 NumPy 배열을 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=" , ")

노트북에서 CSV 파일이 저장된 위치로 이동하면 데이터를 볼 수 있습니다.

예 2: NumPy 배열을 특정 형식의 CSV로 내보내기

숫자의 기본 형식은 “%.18e”입니다. 이는 18개의 0을 표시합니다. 그러나 fmt 인수를 사용하여 다른 형식을 지정할 수 있습니다.

예를 들어 다음 코드는 NumPy 배열을 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 (using 2 decimal places)
n.p. savetxt (" my_data.csv ", data, delimiter=" , ", fmt=" %.2f ")

CSV 파일이 저장된 위치로 이동하면 데이터를 볼 수 있습니다.

예시 3: NumPy 배열을 헤더가 있는 CSV로 내보내기

다음 코드는 NumPy 배열을 사용자 정의 열 헤더가 있는 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 (using 2 decimal places)
n.p. savetxt (" my_data.csv ", data, delimiter=" , ", fmt=" %.2f ",
           header=" A, B, C ", comments="")

참고 : comments 인수는 “#” 기호가 헤더에 표시되는 것을 방지합니다.

CSV 파일이 저장된 위치로 이동하면 데이터를 볼 수 있습니다.

참고 : 여기에서 numpy.savetxt() 함수에 대한 전체 문서를 찾을 수 있습니다.

추가 리소스

다음 튜토리얼에서는 Python에서 다른 일반적인 읽기 및 쓰기 작업을 수행하는 방법을 설명합니다.

NumPy로 CSV 파일을 읽는 방법
Pandas로 CSV 파일을 읽는 방법
Pandas DataFrame을 CSV 파일로 내보내는 방법