텍스트 파일 읽어 하트 모양 워드 클라우드 만들기

창세기1장.txt

0.00MB

heart_mask.png

0.06MB

# genesis1_wordcloud.py

from konlpy.tag import Okt
from wordcloud import WordCloud
import matplotlib.pyplot as plt
from PIL import Image
import numpy as np

# 파일 읽기
with open('창세기1장.txt', 'r', encoding='utf-8') as file:
    genesis_text = file.read()

# 한국어 형태소 분석기 초기화
okt = Okt()

# 명사만 추출
nouns = okt.nouns(genesis_text)
nouns_text = ' '.join(nouns)

# 제공된 하트 모양 마스크 이미지 로드
heart_mask = np.array(Image.open('heart_mask.png'))  # 마스크 이미지 파일 경로를 지정하세요.

# 워드 클라우드 객체 생성
wordcloud = WordCloud(width = 800, height = 800, 
                      background_color ='white', 
                      min_font_size = 10,
                      font_path='C:/Windows/Fonts/malgun.ttf',
                      mask=heart_mask).generate(nouns_text)

# 워드 클라우드 시각화
plt.figure(figsize = (8, 8), facecolor = None) 
plt.imshow(wordcloud, interpolation="bilinear") 
plt.axis("off") 
plt.tight_layout(pad = 0) 

plt.show()

$ python genesis1_wordcloud.py