YAML 파일은 데이터를 인간이 읽을 수 있는 형식으로 저장하는 파일 형식입니다. YAML 파일에는 여러 개의 데이터가 있을 수 있으며, 각 데이터는 키-값 쌍으로 구성되어 있습니다. 때로는 YAML 파일에서 특정 데이터만 필요할 수 있습니다. 이때 PyYAML 라이브러리를 사용하여 YAML 파일의 데이터를 필터링할 수 있습니다.

필요한 라이브러리 설치하기

먼저 PyYAML 라이브러리를 설치해야 합니다. 아래의 명령어를 사용하여 설치할 수 있습니다.

pip install pyyaml

YAML 파일에서 데이터 필터링하기

다음은 PyYAML을 사용하여 YAML 파일에서 데이터를 필터링하는 예제 코드입니다.

import yaml

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

# 필터링할 데이터 설정
filter_key = 'name'

# 데이터 필터링
filtered_data = [data for data in yaml_data if filter_key in data]

# 필터링된 데이터 출력
for data in filtered_data:
    print(data)

위의 코드에서는 data.yaml 파일을 읽어와서 yaml_data에 저장합니다. 필터링할 데이터의 키를 filter_key 변수에 지정합니다. 이후, 리스트 컴프리헨션을 사용하여 filter_key를 포함하는 데이터만을 filtered_data 리스트에 저장합니다. 마지막으로, 필터링된 데이터를 출력합니다.

위의 예제 코드는 name이라는 키를 필터링할 데이터로 설정하였습니다. 필요에 따라 키와 값의 조건을 변경하여 데이터를 필터링할 수 있습니다.

결론

이제 PyYAML을 사용하여 YAML 파일에서 필요한 데이터를 필터링하는 방법을 알게 되었습니다. 이를 통해 YAML 파일에서 필요한 데이터만을 추출하여 다양한 용도로 활용할 수 있습니다.

참고 자료

• PyYAML 공식 문서
• YAML 언어 사양