【Python】WordCloud：mask（マスク）

2021年9月27日2023年3月10日

URLをコピーしました！

WordCloud

前回、WordCloudのオプションを色々試してみて、使い方を勉強してみました。

テキストファイルの読み込みと形態素解析

まずはテキストファイルを読み込み、形態素解析をしましょう。

先ほどの「data.txt」をプログラムと同じフォルダに配置します。

そしてjanomeの解説の際に覚えた「分かち書き」をこちらのプログラムで行います。

from janome.tokenizer import Tokenizer

with open("./data.txt", "r") as f:
    data = f.read()

tk = Tokenizer(wakati=True)
tokens = tk.tokenize(data)

words = " ".join(list(tokens))

print(words)

前回と違うのはこちらのデータの読み込みの部分だけです。

with open("./data.txt", "r") as f:
    data = f.read()

テキストファイルを読み込むのによくやるのは、for文を使って1行ずつ読み込むことですが、janomeには一度に渡してしまえるため、「data = f.read()」でテキストファイル全文をそのままdataに格納しています。

あとは前と同様にjanomeのtokenaizerで分かち書きに変換しています。

WordCloudのmaskを試す

それではWordCloudのmaskを試してみましょう。

プログラムとしてはこんな感じになります。

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

img = Image.open("./py_img.png")
mask_img = np.array(img)

wordcloud = WordCloud(mask=mask_img,font_path=r"/System/Library/Fonts/Hiragino Sans GB.ttc").generate(words)
wordcloud.to_file("./sample.png")

まず画像をPIL（Python Image Library）で読み込みます。

img = Image.open("./py_img.png")

PILに関しては前に解説していますので、良かったらこちらの記事をどうぞ。

あわせて読みたい

【Pillow（PIL）】画像のトリミング、サイズ変更、保存[Python] 【Pillow（PIL）】ブログを運営していて煩雑になるのは、画像の処理。大きいままだとサーバーの容量を逼迫してしまうし、小さすぎたり、解像度が低すぎたりすると画像…

そして読み込んだ画像ファイルをnumpyを使ってndarray形式に変換します。

mask_img = np.array(img)

そしてndarray形式にした画像データをmaskの引数に入れて、ワードクラウドを作成します。

wordcloud = WordCloud(mask=mask_img,font_path=r"/System/Library/Fonts/Hiragino Sans GB.ttc").generate(words)
wordcloud.to_file("./sample.png")

これで実行してみるとこんな画像が出力されます。

つまりmaskとは、読み込んだ画像の形に合わせてワードクラウドを作ってくれる機能なのです。

背景が黒ではちょっとパッとしないので、白にしてみました。

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

img = Image.open("./py_img.png")
mask_img = np.array(img)

wordcloud = WordCloud(mask=mask_img, background_color="white", font_path=r"/System/Library/Fonts/Hiragino Sans GB.ttc").generate(words)
wordcloud.to_file("./sample.png")

実行結果

やはり背景は白の方が映えますね。

contour_widthを試す

では次にcontour_widthを試してみましょう。

これはmaskと併せて使う機能なので、先ほどの背景を白にしたプログラムにcontour_widthを追加します。

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

img = Image.open("./py_img.png")
mask_img = np.array(img)

wordcloud = WordCloud(mask=mask_img, background_color="white", contour_width=5, font_path=r"/System/Library/Fonts/Hiragino Sans GB.ttc").generate(words)
wordcloud.to_file("./sample.png")

実行結果

枠線が表示されました。

ちなみにcontour_widthは数字（int）で指定しますが、その数字が枠線の太さを示しています。

contour_colorを試す

ここまで来れば、contour_colorは何か分かるかと思います。

そう、枠線の色です。

枠線を表示した先ほどのプログラムにcontour_colorを追加して枠線の色を変えてみましょう。

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

img = Image.open("./py_img.png")
mask_img = np.array(img)

wordcloud = WordCloud(mask=mask_img, background_color="white", contour_color="red", contour_width=2, font_path=r"/System/Library/Fonts/Hiragino Sans GB.ttc").generate(words)
wordcloud.to_file("./sample.png")

実行結果