Python 나무위키 데이터 가져오기 (Pandas, datasets, parquet)

안녕하세요.

 

오늘은 나무위키에 있는 '식품 관련 정보' 데이터를 가져오는 걸 구현 해보겠습니다.

https://huggingface.co/datasets/heegyu/namuwiki 를 참고 하였습니다.

 

 

datasets 을 설치하고, namuwiki 데이터 즉, parquet 데이터를 다운로드 받습니다.

 

$ pip install datasets

from datasets import load_dataset("heegyu/namuwiki")


dataset = load_dataset("heegyu/namuwiki")

수 많은 데이터가 존재하고, 거기서 본인이 원하는 데이터를 아래와 같이 검색하면 됩니다.

 

import pandas as pd


df = pd.read_parquet("/Users/user/Downloads/namuwiki_20210301.parquet")
filtered_df = df[df.str.contains("식품 관련 정보")]
filtered_df['text'].to_csv("/Users/user/Downloads/info.csv")

 

그러면 아래와 같은 데이터를 확인할 수 있습니다.

 

감사합니다.

 

참고

https://huggingface.co/datasets/heegyu/namuwiki