from wordcloud import WordCloud import matplotlib.pyplot as plt def get_word_cloud(words_list): #首先實例化詞雲類對象,裏面三個參數 #font_path:字體路徑,爲了能夠更好的顯示中文 #max_words:指定詞雲圖像最多可以顯示的詞彙數量 #backgroud_color:代表圖片的背景顏色 wordcloud = WordCloud(width=300,height=200,font_path="SimHei.ttf",max_words=100,background_color="white") #將傳入的列表參數轉化爲字符串形式,因爲詞雲對象的參數要求是字符串類型 keywords_string = " ".join(words_list) #生成詞雲 wordcloud.generate(keywords_string) #繪圖 plt.figure() plt.imshow(wordcloud,interpolation="bilinear") plt.axis("off") plt.show() if __name__ == '__main__': words = "高級AI工程師 劉新宇 liuxinyu nuttertools xinyuuliu xinyuliudr liuxy 算法工程師 自然語言處理 圖像處理 seq2seq pytorch tensorflow YOLO BERT LSTM transformer GPT".split() get_word_cloud(words)
安裝詞雲 pip install wordcloud
在自然語言處理裏邊,可以顯示出一些高頻詞
效果如下,列表越排在前面的字符,顯示的字符越大,越排在後面的顯示的字符越小