기타

npy 파일 읽기 & pickle 에러 해결

남하욱 2024. 5. 21. 15:43

data를 사용할 때, npy형식인 data들이 있다. 이는 numpy에서 numpy.ndarray 형식 data를 저장한 것이다.

 

npy형식 파일들은 다음과 같이 np.load 함수를 통해 읽을 수 있다.

import numpy as np

data = np.load("my_data.npy")

또한 이렇게 npy 파일을 load하는 과정에서 다음과 같은 error 메세지가 뜨는 경우가 있다.

ValueError: Object arrays cannot be loaded when allow_pickle=False

 

이 경우 다음과 같이 위의 np.load 함수에 pickle=True를 넣어주면 된다.

import numpy as np

data = np.load("my_data.npy", allow_pickle=True)