Numpy 배열 저장과 불러오기

Numpy 라이브러리를 사용하면 배열 데이터를 파일로 저장하거나 파일에서 불러올 수 있습니다. 이를 위해 np.save()와 np.savez() 함수를 사용할 수 있습니다.

 

1. 단일  Numpy배열 저장하기

np.save('경로/파일명.npy', numpy데이터)

  - save 함수는 배열을 1개만 저장할 수 있습니다. 
  - save 시 확장자는 .npy 를 사용합니다.

 

array = [1, 3, 5, 7, 9, 2, 4, 6, 8, 0]

np.save('array.npy', array)
new_array = np.load('array.npy')
new_array

파일이 생성된 모습

 

2. 복수 Numpy 저장하기

np.savez('경로/파일명.npz', numpy데이터 = 이름 , ---)

  - savez 함수는 여러 개의 배열을 저장할 수 있습니다.

  - savez 시 확장자는 .npz를 사용합니다.

arr1 = ['a', 'b', 'c', 'd', 'e']
arr2 = [1, 2, 3, 4, 5]
arr3 = ['ㄱ', 'ㄴ', 'ㄷ', 'ㄹ', 'ㅂ']

np.savez('arrays.npz', arr1=arr1, arr2=arr2, arr3=arr3) # 각 배열에 이름 부여

파일이 생성된 모습

savez의 경우 딕셔너리 형태로 지정된 이름을 key값을 통해 numpy 배열을 가져올 수 있습니다.

new_arrays = np.load('arrays.npz')
print(new_arrays['arr1'])
print(new_arrays['arr2'])
print(new_arrays['arr3'])

 

감사합니다.

 

-sjkoding-