wordcloud用来制作词云
生活随笔
收集整理的這篇文章主要介紹了
wordcloud用来制作词云
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
也許你會看到wordcloud和wordcloud2,這里區別一下,wordcloud是python庫,wordcloud2是R語言的庫,我用的python,所以下面的例子都是基于wordcloud演示的。
github:https://github.com/amueller/word_cloud
官方地址:https://amueller.github.io/word_cloud/
下載.whl文件http://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud
使用cd命令進入whl文件的路徑
運行這條命令:
[python]?view plaincopy
自定義字體顏色
這段代碼主要來自wordcloud的github,你可以在github下載該例子
#!/usr/bin/env python """ Colored by Group Example ========================Generating a word cloud that assigns colors to words based on a predefined mapping from colors to words """from wordcloud import (WordCloud, get_single_color_func) import matplotlib.pyplot as pltclass SimpleGroupedColorFunc(object):"""Create a color function object which assigns EXACT colorsto certain words based on the color to words mappingParameters----------color_to_words : dict(str -> list(str))A dictionary that maps a color to the list of words.default_color : strColor that will be assigned to a word that's not a memberof any value from color_to_words."""def __init__(self, color_to_words, default_color):self.word_to_color = {word: colorfor (color, words) in color_to_words.items()for word in words}self.default_color = default_colordef __call__(self, word, **kwargs):return self.word_to_color.get(word, self.default_color)class GroupedColorFunc(object):"""Create a color function object which assigns DIFFERENT SHADES ofspecified colors to certain words based on the color to words mapping.Uses wordcloud.get_single_color_funcParameters----------color_to_words : dict(str -> list(str))A dictionary that maps a color to the list of words.default_color : strColor that will be assigned to a word that's not a memberof any value from color_to_words."""def __init__(self, color_to_words, default_color):self.color_func_to_words = [(get_single_color_func(color), set(words))for (color, words) in color_to_words.items()]self.default_color_func = get_single_color_func(default_color)def get_color_func(self, word):"""Returns a single_color_func associated with the word"""try:color_func = next(color_func for (color_func, words) in self.color_func_to_wordsif word in words)except StopIteration:color_func = self.default_color_funcreturn color_funcdef __call__(self, word, **kwargs):return self.get_color_func(word)(word, **kwargs)text = """The Zen of Python, by Tim Peters Beautiful is better than ugly. Explicit is better than implicit. Simple is better than complex. Complex is better than complicated. Flat is better than nested. Sparse is better than dense. Readability counts. Special cases aren't special enough to break the rules. Although practicality beats purity. Errors should never pass silently. Unless explicitly silenced. In the face of ambiguity, refuse the temptation to guess. There should be one-- and preferably only one --obvious way to do it. Although that way may not be obvious at first unless you're Dutch. Now is better than never. Although never is often better than *right* now. If the implementation is hard to explain, it's a bad idea. If the implementation is easy to explain, it may be a good idea. Namespaces are one honking great idea -- let's do more of those!"""# Since the text is small collocations are turned off and text is lower-cased wc = WordCloud(collocations=False).generate(text.lower())# 自定義所有單詞的顏色 color_to_words = {# words below will be colored with a green single color function'#00ff00': ['beautiful', 'explicit', 'simple', 'sparse','readability', 'rules', 'practicality','explicitly', 'one', 'now', 'easy', 'obvious', 'better'],# will be colored with a red single color function'red': ['ugly', 'implicit', 'complex', 'complicated', 'nested','dense', 'special', 'errors', 'silently', 'ambiguity','guess', 'hard'] }# Words that are not in any of the color_to_words values # will be colored with a grey single color function default_color = 'grey'# Create a color function with single tone # grouped_color_func = SimpleGroupedColorFunc(color_to_words, default_color)# Create a color function with multiple tones grouped_color_func = GroupedColorFunc(color_to_words, default_color)# Apply our color function # 如果你也可以將color_func的參數設置為圖片,詳細的說明請看 下一部分 wc.recolor(color_func=grouped_color_func)# Plot plt.figure() plt.imshow(wc, interpolation="bilinear") plt.axis("off") plt.show()利用背景圖片生成詞云,設置停用詞詞集
該段代碼主要來自于wordcloud的github,你同樣可以在github下載該例子以及原圖片與效果圖
#!/usr/bin/env python """ Image-colored wordcloud =======================You can color a word-cloud by using an image-based coloring strategy implemented in ImageColorGenerator. It uses the average color of the region occupied by the word in a source image. You can combine this with masking - pure-white will be interpreted as 'don't occupy' by the WordCloud object when passed as mask. If you want white as a legal color, you can just pass a different image to "mask", but make sure the image shapes line up. """from os import path from PIL import Image import numpy as np import matplotlib.pyplot as pltfrom wordcloud import WordCloud, STOPWORDS, ImageColorGeneratord = path.dirname(__file__)# Read the whole text. text = open(path.join(d, 'alice.txt')).read()# read the mask / color image taken from # http://jirkavinse.deviantart.com/art/quot-Real-Life-quot-Alice-282261010 alice_coloring = np.array(Image.open(path.join(d, "alice_color.png")))# 設置停用詞 stopwords = set(STOPWORDS) stopwords.add("said")# 你可以通過 mask 參數 來設置詞云形狀 wc = WordCloud(background_color="white", max_words=2000, mask=alice_coloring,stopwords=stopwords, max_font_size=40, random_state=42) # generate word cloud wc.generate(text)# create coloring from image image_colors = ImageColorGenerator(alice_coloring)# show # 在只設置mask的情況下,你將會得到一個擁有圖片形狀的詞云 plt.imshow(wc, interpolation="bilinear") plt.axis("off") plt.figure() # recolor wordcloud and show # we could also give color_func=image_colors directly in the constructor # 我們還可以直接在構造函數中直接給顏色 # 通過這種方式詞云將會按照給定的圖片顏色布局生成字體顏色策略 plt.imshow(wc.recolor(color_func=image_colors), interpolation="bilinear") plt.axis("off") plt.figure() plt.imshow(alice_coloring, cmap=plt.cm.gray, interpolation="bilinear") plt.axis("off") plt.show()展示效果如下:
創作挑戰賽新人創作獎勵來咯,堅持創作打卡瓜分現金大獎
總結
以上是生活随笔為你收集整理的wordcloud用来制作词云的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python中文分词jieba总结
- 下一篇: wordcloud用来制作中文词云