# 導(dǎo)入模塊
from wordcloud import WordCloud
# 準(zhǔn)備文本數(shù)據(jù)，是一個(gè)字符串，單詞之間用空格分隔
sentence = "hello satori hello mashiro hello satori"
# 創(chuàng)建詞云對(duì)象
wc = WordCloud()
# 根據(jù)文本生成詞云
wc.generate(sentence)
# 保存為圖片
wc.to_file("word.png")

from wordcloud import WordCloud
sentence = "i do not need sex, because life fucks me every every single day"
wc = WordCloud(
    width=500,  # 設(shè)置寬度為500px
    height=300,  # 設(shè)置高度為300px
    background_color='pink',  # 設(shè)置背景為粉色
    stopwords={"sex", "fucks"},  # 設(shè)置禁用詞
    max_font_size=100,  # 設(shè)置最大的字體大小，所有詞都不會(huì)超過 100px
    min_font_size=10,  # 設(shè)置最小的字體大小，所有詞都會(huì)超過 10px
    max_words=10  # 最多生成 10 個(gè)詞，當(dāng)然這里單詞比較少，看不出來什么
)

wc.generate(sentence)
wc.to_file("word.png")

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

# 一篇英文文章
with open("article.txt") as f:
    sentence = f.read()
# 加載一張圖片，轉(zhuǎn)化成numpy中的數(shù)組
mask = np.array(Image.open("哆啦A夢(mèng).png"))
# 傳入mask
wc = WordCloud(mask=mask)
wc.generate(sentence)
wc.to_file("word.png")

from wordcloud import WordCloud

wc = WordCloud()
wc.generate("古明地覺的編程教室")
wc.to_file("word.png")

from wordcloud import WordCloud

# 傳入本機(jī)支持中文的字體名稱
wc = WordCloud(font_path="Arial Unicode.ttf")
wc.generate("古明地覺的編程教室")
wc.to_file("word.png")

from wordcloud import WordCloud
import jieba

with open("出師表.txt") as f:
    sentence = f.read()
# 分詞得到列表，手動(dòng)使用空格拼接
sentence = " ".join(jieba.cut(sentence))
wc = WordCloud(font_path="Arial Unicode.ttf")
wc.generate(sentence)
wc.to_file("word.png")

from io import BytesIO
from wordcloud import WordCloud
import jieba

with open("出師表.txt") as f:
    sentence = f.read()

sentence = " ".join(jieba.cut(sentence))
wc = WordCloud(font_path="Arial Unicode.ttf")
wc.generate(sentence)

# 將詞云保存為 PIL 的 Image 對(duì)象
im = wc.to_image()
buf = BytesIO()
# 將詞云的字節(jié)流保存在 buf 中，這樣可以直接交給客戶端進(jìn)行渲染
im.save(buf, "png")
print(buf.getvalue())

# 當(dāng)然也可以保存為文件，im.save(filename)
# wc.to_file() 底層也是先轉(zhuǎn)成 Image 對(duì)象、然后調(diào)用 im.save() 實(shí)現(xiàn)的

# 或者還可以保存為 SVG 格式
svg = wc.to_svg()
# 將 svg 的內(nèi)容保存成文件，就得到 SVG 圖片了
print(svg)