[python] Matplotlib로 워드클라우드 그리기

Matplotlib은 파이썬의 시각화 라이브러리 중 하나로, 다양한 그래프와 플롯을 그릴 수 있습니다. 이번 글에서는 Matplotlib를 사용하여 워드클라우드를 그리는 방법에 대해 알아보겠습니다.

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

먼저, 워드클라우드를 그리기 위해 필요한 라이브러리를 설치해야 합니다. 다음 명령어를 사용하여 필요한 라이브러리를 설치할 수 있습니다.

pip install matplotlib
pip install wordcloud

워드클라우드 데이터 준비하기

Matplotlib를 사용하여 워드클라우드를 그리기 위해서는 워드클라우드에 표시할 데이터를 준비해야 합니다. 예를 들어, 문자열 리스트 형태의 데이터를 사용할 수 있습니다.

data = ['apple', 'banana', 'orange', 'apple', 'pear', 'grape', 'banana']

파일로부터 데이터를 읽어올 수도 있습니다.

with open('data.txt', 'r') as f:
    data = f.read().split()

워드클라우드 생성하기

Matplotlib의 pyplot 모듈을 사용하여 워드클라우드를 생성할 수 있습니다. 다음은 간단한 예제 코드입니다.

import matplotlib.pyplot as plt
from wordcloud import WordCloud

def create_wordcloud(data):
    wordcloud = WordCloud().generate(' '.join(data))
    plt.imshow(wordcloud, interpolation='bilinear')
    plt.axis('off')
    plt.show()

create_wordcloud(data)

위 코드에서는 WordCloud() 함수를 사용하여 워드클라우드 객체를 생성하고, generate() 메서드를 사용하여 데이터를 워드클라우드에 적용시킵니다. 그리고 imshow() 함수를 사용하여 이미지를 표시하고, axis('off')를 통해 축을 숨깁니다. 마지막으로 show() 함수를 호출하여 워드클라우드를 화면에 표시합니다.

결과 확인하기

위 코드를 실행하면, 해당 데이터를 기반으로 워드클라우드 이미지가 생성되어 화면에 표시됩니다. 워드클라우드의 크기나 색상 등을 지정하는 옵션도 추가할 수 있습니다.

참고 자료