[Python] npz 파일 키 확인하는 법

이전 포스팅에서 npy, npz 파일을 만들고 여는 법을 살펴봤다.

이번 포스팅에서는 npz 파일의 키를 확인하는 법을 알아본다.

---

npz 파일 key 확인하기

files 속성을 사용하여 아래와 같은 코드로 키를 확인할 수 있다.

import numpy as np

# .npz 파일 열기
data = np.load('파일이름.npz')

# 키 확인하기
keys = data.files
print(keys)

---

예제

npz 파일을 열고 키를 확인한 다음 첫 번째 키에 해당하는 배열을 출력해보자.

import numpy as np

# .npz 파일 열기
data = np.load('example.npz')

# 키 확인하기
keys = data.files
print("Keys:", keys)

# 첫 번째 키에 해당하는 배열 출력하기
array = data[keys[0]]
print(array)

# 파일 닫기
data.close()

with문 사용

참고로 with문을 사용하면 파일을 닫는 과정이 필요가 없다.

자동으로 닫힌다.

import numpy as np

# with 문을 사용하여 .npz 파일 열기
with np.load('example.npz') as data:
    # 키 확인하기
    keys = data.files
    print("Keys:", keys)

    # 첫 번째 키에 해당하는 배열 출력하기
    array = data[keys[0]]
    print(array)

# with 문을 사용하면 파일이 자동으로 닫힙니다.