pyvista.Plotter.add_legend

pyvista.Plotter.add_legend#

Plotter.add_legend(
labels=None,
bcolor=None,
border=False,
size=(0.2, 0.2),
name=None,
loc='upper right',
face=None,
font_family=None,
background_opacity=1.0,
)[source]#

Add a legend to render window.

Entries must be a list containing one string and color entry for each item.

Parameters:
labelslist, optional

When set to None, uses existing labels as specified by

List containing one entry for each item to be added to the legend. Each entry can contain one of the following:

  • Two strings ([label, color]), where label is the name of the item to add, and color is the color of the label to add.

  • Three strings ([label, color, face]) where label is the name of the item to add, color is the color of the label to add, and face is a string which defines the face (i.e. circle, triangle, box, etc.). face could be also "none" (no face shown for the entry), or a pyvista.PolyData.

  • A dict with the key label. Optionally you can add the keys color and face. The values of these keys can be strings. For the face key, it can be also a pyvista.PolyData.

bcolorColorLike, default: (0.5, 0.5, 0.5)

Background color, either a three item 0 to 1 RGB color list, or a matplotlib color string (e.g. 'w' or 'white' for a white color). If None, legend background is disabled.

borderbool, default: False

Controls if there will be a border around the legend. Default False.

sizesequence[float], default: (0.2, 0.2)

Two float sequence, each float between 0 and 1. For example (0.1, 0.1) would make the legend 10% the size of the entire figure window.

namestr, optional

The name for the added actor so that it can be easily updated. If an actor of this name already exists in the rendering window, it will be replaced by the new actor.

locstr, default: “upper right”

Location string. One of the following:

  • 'upper right'

  • 'upper left'

  • 'lower left'

  • 'lower right'

  • 'center left'

  • 'center right'

  • 'lower center'

  • 'upper center'

  • 'center'

facestr | pyvista.PolyData, optional

Face shape of legend face. Defaults to a triangle for most meshes, with the exception of glyphs where the glyph is shown (e.g. arrows).

You may set it to one of the following:

  • None: "none"

  • Line: "-" or "line"

  • Triangle: "^" or 'triangle'

  • Circle: "o" or 'circle'

  • Rectangle: "r" or 'rectangle'

  • Custom: pyvista.PolyData

Passing "none" removes the legend face. A custom face can be created using pyvista.PolyData. This will be rendered from the XY plane.

font_familystr, optional

Font family. Must be either 'courier', 'times', or 'arial'. Defaults to pyvista.global_theme.font.family.

background_opacityfloat, default: 1.0

Set background opacity.

Returns:
vtk.vtkLegendBoxActor

Actor for the legend.

Examples

Create a legend by labeling the meshes when using add_mesh

>>> import pyvista as pv
>>> from pyvista import examples
>>> sphere = pv.Sphere(center=(0, 0, 1))
>>> cube = pv.Cube()
>>> plotter = pv.Plotter()
>>> _ = plotter.add_mesh(sphere, 'grey', smooth_shading=True, label='Sphere')
>>> _ = plotter.add_mesh(cube, 'r', label='Cube')
>>> _ = plotter.add_legend(bcolor='w', face=None)
>>> plotter.show()
../../../_images/pyvista-Plotter-add_legend-1_00_00.png

Alternatively provide labels in the plotter.

>>> plotter = pv.Plotter()
>>> _ = plotter.add_mesh(sphere, 'grey', smooth_shading=True)
>>> _ = plotter.add_mesh(cube, 'r')
>>> legend_entries = []
>>> legend_entries.append(['My Mesh', 'w'])
>>> legend_entries.append(['My Other Mesh', 'k'])
>>> _ = plotter.add_legend(legend_entries)
>>> plotter.show()
../../../_images/pyvista-Plotter-add_legend-1_01_00.png