词云

from wordcloud import WordCloud
import matplotlib.pyplot as plt
import jieba

def en_wordcloud():
    with open('1.txt', 'r') as f:
        mytext = f.read()
        print(mytext)

    wordcloud = WordCloud().generate(mytext)

    plt.imshow(wordcloud, interpolation='bilinear')
    plt.axis('off')
    plt.show()


def zh_wordcloud():
    with open('2.txt', 'r', encoding='utf-8') as f:
        text = f.read()

    cut_text = "".join(jieba.cut(text))

    cloud = WordCloud(
        font_path='c:\windows\fonts\stxingka.ttf',
        background_color='white',
        max_words=4000,
        max_font_size=60
    )

    wCloud = cloud.generate(cut_text)
    wCloud.to_file('cloud.jpg')

    plt.imshow(wCloud, interpolation='bilinear')
    plt.axis('off')
    plt.show()


if __name__ == '__main__':
    # en_wordcloud()
    zh_wordcloud()

image

posted @ 2021-10-08 13:41  guangzan  阅读(74)  评论(0)    收藏  举报