Yellowbrick 更改图例并添加标题

Question

我用 Yellowbrick RadViz 创建了一个图表：visualizer = RadViz(classes=labels)visualizer.fit(X, y) visualizer.transform(X)  visualizer.show()如您所见，图例覆盖了一些功能名称：此外，我想编辑标题。我试过：visualizer.ax.set_title("new title")visualizer.fig.legend(bbox_to_anchor=(1.02, 1), loc=0, borderaxespad=0., title = "level")但是set_title 没有效果。使用fig.legend ，确实创建了一个新图例，但我无法删除原始图例。怎么做到呢？

Answer 1

您可以使用该参数修改 Yellowbrick 图的标题title，并使用该size参数来增加轴的大小，这可能有助于重叠标签。大小被指定为像素尺寸的元组：

from yellowbrick.features import RadViz

from yellowbrick.datasets import load_occupancy

X, y = load_occupancy()

visualizer = RadViz(

    classes=["occupied", "vacant"], 

    title="My custom title", 

    size=(800, 600)

)

visualizer.fit(X, y)

visualizer.transform(X)

visualizer.show()

或者，可以通过绕过可视化工具show()和finalize()方法来跳过添加 Yellowbrick 图例和标题的步骤，然后使用绘图所需的任何自定义图例位置直接修改 ax 对象：

from yellowbrick.features import RadViz

from yellowbrick.datasets import load_occupancy

X, y = load_occupancy()

visualizer = RadViz()

visualizer.fit(X, y)

visualizer.transform(X)

custom_viz = visualizer.ax

custom_viz.set_title("New title")

custom_viz.figure.legend(

    bbox_to_anchor=(1.02, 1), 

    borderaxespad=0.0,

    title="level",

    loc=0,

)

custom_viz.figure.show()