我是python中的新手,我使用的是wordcloud pkg。
我按照dict frequencies
中的频率生成了一个词云,其中keys = words和values =单词的频率。
wordcloud可以在color_funct
参数中接收函数。此功能可用于为云上的单词制作自己的色彩映射。例如,我使用my_tf_color_func
。
wordcloud = WordCloud(background_color="white", max_words=1000, mask=motog3_coloring,
max_font_size=1000, random_state=1, relative_scaling=.6, regexp="\w[\%\-\(\)\w']+",
collocations=False, color_func=my_tf_color_func)
在my_tf_color_func
中,我想根据属性frequencies
中的频率为每个单词着色。鉴于报告on this issue
my_tf_color_func
中执行类似下面的操作
def my_tf_color_func(word, font_size, position, orientation, random_state=None, **kwargs):
return "hsl(%d, 80%%, 50%%)" % (360 * frequencies[word])
但必须传递dict frequencies
,我该怎么做?我必须在哪里以及如何通过词典frequencies
才能在my_tf_color_func
中使用它。
我试过某种方式,但我做不到。
非常感谢
答案 0 :(得分:2)
您可以使用以下内容明确地关闭它:
def my_tf_color_func(dictionary):
def my_tf_color_func_inner(word, font_size, position, orientation, random_state=None, **kwargs):
return "hsl(%d, 80%%, 50%%)" % (360 * dictionary[word])
return my_tf_color_func_inner
wordcloud = WordCloud(...,color_func=my_tf_color_func(frequencies))
显式有状态函数类对象的替代方法如下:
class MyColorFunctor():
def __init__(self,frequencies):
self.frequencies = frequencies
def __call__(self,word,font_size,position,orientation,random_state=None,**kwargs):
return "hsl(%d, 80%%, 50%%)" % (360 * self.frequencies[word])
wordcloud = WordCloud(...,color_func=MyColorFunctor(frequencies))