카카오톡 워드클라우드 만들기

워드클라우드 환경 설정

워드클라우드 패키지를 설치한다.

File > Settings > Python Interpreter > wordcloud

import matplotlib.font_manager as fm

# 이용 가능한 폰트 중 '고딕'만 선별 -> 글씨가 굵어야 예뻐서
for font in fm.fontManager.ttflist:
    if 'Gothic' in font.name:
        print(font.name, font.fname)

이 코드를 실행시키면 그러면 폰트의 경로들이 쭉 출력될 텐데,

그중 마음에 드는 폰트 경로 하나 복사해놓기.

(BUT 워드클라우드가 제대로 만들어지지 않는 폰트들이 많기 때문에

MalgunGothic(윈도우), AppleGothic(맥)을 주로 사용하자!)

 

from wordcloud import WordCloud
text = ''
with open("KakaoTalk.txt", "r", encoding="utf-8") as f:
    lines = f.readlines()
    for line in lines:
        text += line

print(text)

wc = WordCloud(font_path="C:/windows/Fonts/malgunbd.ttf", background_color="white", width=600, height=400)
wc.generate(text)
wc.to_file("result.png")

카카오톡 내보내기를 통해 만들어진 txt 파일을 넣어주면 워드 클라우드가 생성된다.

 

(차마 내용을 공개할 수 없었습니다💦)

여하튼 워드클라우드를 확인해보면, 상대방 이름이나 오후 시간 등 쓸데없는 정보가 많이 포함되어 있는 것을 알 수 있다.

 

데이터 클렌징

데이터 클렌징은 불필요한 데이터를 제거하는 과정이다.

from wordcloud import WordCloud
text = ''
with open("KakaoTalk.txt", "r", encoding="utf-8") as f:
    lines = f.readlines()
    for line in lines[5:]:
        if'] [' in line:
            text += line.split('] ')[2].replace('ㅋ','').replace('ㅠ','').replace('이모티콘\n','').replace('사진\n','').replace('삭제된 메시지입니다.', '')

print(text)

wc = WordCloud(font_path="C:/windows/Fonts/malgunbd.ttf", background_color="white", width=600, height=400)
wc.generate(text)
wc.to_file("result.png")

시스템 메시지와 보낸 사람 이름, 보낸 시각 등을 삭제하였고

그 외 이모티콘이나 ㅋㅋㅋ, ㅠㅠㅠㅠㅠ같은 것들도 replace로 지웠다.

 

워드클라우드 모양 만들기

이런 이미지처럼, 배경과 구분이 확실한 이미지가 모양이 잘 만들어진다.

from wordcloud import WordCloud
from PIL import Image
import numpy as np

text = ''
with open("KakaoTalk.txt", "r", encoding="utf-8") as f:
    lines = f.readlines()
    for line in lines[5:]:
        if'] [' in line:
            text += line.split('] ')[2].replace('ㅋ','').replace('ㅠ','').replace('이모티콘\n','')\
                .replace('사진\n','').replace('삭제된 메시지입니다.', '')


print(text)

mask = np.array(Image.open('cloud.png'))
wc = WordCloud(font_path="C:/windows/Fonts/malgunbd.ttf", background_color="white", mask=mask)
wc.generate(text)
wc.to_file("result_masked.png")

그럼 이렇게 구름모양의 워드 클라우드가 나온다!

(다시 한번 말하지만, 차마 내용을 공개할 수 없었습니다..)