Python wordcloud 颜色按词频

Question

我是 python 的新手，我正在使用 wordcloud pkg。

我根据字典 frequencies 中的频率生成了一个词云，key=words 和 values=frequencies of the words。

词云可以在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 中每个词的出现频率为每个词着色。我现在可以在 my_tf_color_func 中执行如下操作，给出报告 on this issue

def my_tf_color_func(word, font_size, position, orientation, random_state=None, **kwargs):
         return "hsl(%d, 80%%, 50%%)" % (360 * frequencies[word])

但是字典 frequencies 必须被传递，我该怎么做？我必须在哪里以及如何传递字典 frequencies 才能在 my_tf_color_func 中使用它。

我尝试了一些方法，但我做不到。

非常感谢

Answer 1

您可以使用如下方式显式关闭它:

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