按术语频率划分的Python wordcloud颜色

时间:2017-11-06 18:32:51

标签: python kwargs word-cloud

我是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中使用它。

我试过某种方式,但我做不到。

非常感谢

1 个答案:

答案 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))