代码之家 › 专栏 › 技术社区 › Cranjis

黄砖更改图例并添加标题

yellowbrick data-visualization matplotlib python

0

Cranjis · 技术社区 · 5 年前

我用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 ,确实创建了一个新的传说,但我无法删除原始传说。

如何做到这一点?

0 回复 | 直到 5 年前

1

7

rebeccabilbro 5 年前

您可以使用以下命令修改黄砖地块的标题 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()

或者,可以通过绕过可视化工具来跳过添加Yellowbrick图例和标题的步骤 show() 和 finalize() 方法,然后使用绘图所需的任何自定义图例位置直接修改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()