matplotlib - 更改使用黄砖创建的分散图中的标题

Question

我想更改使用黄砖创建的分散图的标题：

我使用以下代码：

wl = []
with open('my-text.txt', 'r', encoding='utf8') as f:
   wl = f.read().split()
   
topics = ['δὲ']
plt.figure(figsize=(16,.5))
visualizer = DispersionPlot(topics)
visualizer.fit([poll_wl])
visualizer.show()

创建的图的标题为“词汇散布图”。我想使用自定义标题。

标签： matplotlibyellowbrick

Answer 1

您可以figure从 Visalizer 中获取，然后设置唯一axes对象的标题文本，如下所示：

visualizer.fig.get_axes()[0].title.set_text("My Title")

visualizer.set_title("My Title")尽管它似乎没有明确包含在docs中，但也可以这样做。

示例（来自文档）：

from yellowbrick.text import DispersionPlot
from yellowbrick.datasets import load_hobbies

corpus = load_hobbies()
text = [doc.split() for doc in corpus.data]
target_words = ['Game', 'player', 'score', 'oil', 'Man']
visualizer = DispersionPlot(target_words)
visualizer.fit(text)
visualizer.show()

visualizer.set_title("My Title")