[python] PyYAML을 이용하여 YAML 파일에서 특정 데이터를 찾는 방법

YAML은 인간이 쉽게 읽고 작성할 수 있는 데이터 직렬화 양식으로, 다양한 프로그래밍 언어에서 데이터를 저장 및 전달하기 위해 사용됩니다. Python에서는 PyYAML 라이브러리를 사용하여 YAML 파일을 읽고 쓸 수 있습니다. 이번 글에서는 PyYAML을 사용하여 YAML 파일에서 특정 데이터를 찾는 방법에 대해 알아보겠습니다.

PyYAML 설치

먼저, PyYAML을 설치해야 합니다. 아래의 명령어를 사용하여 PyYAML을 설치할 수 있습니다.

pip install pyyaml

YAML 파일 읽기

다음은 PyYAML을 사용하여 YAML 파일을 읽는 예제입니다. 예제로 사용할 YAML 파일은 다음과 같습니다.

# example.yaml
name: John Doe
age: 30
email: john.doe@example.com
address:
  street: 123 Main St
  city: Anytown
  state: NY

아래의 코드를 사용하여 YAML 파일을 읽고 데이터를 가져올 수 있습니다.

import yaml

# YAML 파일 열기
with open('example.yaml', 'r') as file:
    data = yaml.safe_load(file)

# 데이터 확인
print(data)

코드를 실행하면 다음과 같이 YAML 파일의 내용이 출력됩니다.

{'name': 'John Doe', 'age': 30, 'email': 'john.doe@example.com', 'address': {'street': '123 Main St', 'city': 'Anytown', 'state': 'NY'}}

특정 데이터 찾기

이제 YAML 파일에서 특정 데이터를 찾는 방법에 대해 알아보겠습니다. 다음은 name 항목의 값을 가져오는 예제입니다.

name = data['name']
print(name)  # John Doe

한편, 중첩된 데이터의 경우에는 점 표기법을 사용하여 값을 가져올 수 있습니다. 예를 들어, address 항목의 street 값을 가져오는 예제는 다음과 같습니다.

street = data['address']['street']
print(street)  # 123 Main St

위와 같은 방식으로, PyYAML을 사용하여 YAML 파일에서 특정 데이터를 찾을 수 있습니다. 데이터를 찾을 때에는 YAML 파일의 구조에 따라 적절한 접근 방법을 사용하는 것이 중요합니다.

참고 자료