построение нескольких сеток с помощью pyvista

Question

У меня есть несколько сеток, я каждую из них читаю с помощью pyvista

import pyvista as pv

# read the data
grid1 = pv.read('mesh1.vtk')
grid2 = pv.read('mesh2.vtk')

Я хочу построить их вместе на одном участке, используя разные цвета:

plotter = pv.Plotter(window_size=(1500, 1100))
plotter.add_mesh(grid1, color=[0.6, 0.2, 0.1])
plotter.add_mesh(grid2, color=[0.1, 0.6, 0.6])

Могу ли я добавить метку для каждой сетки? или добавить легенду?

Answer 1

Да, это естественно, и вы уже знаете, какие ключевые слова нужно использовать: label а также legend:

import pyvista as pv 
from pyvista import examples 
 
unstructured = examples.load_hexbeam() 
poly = examples.load_ant() 
poly.points /= 10 
poly.points += [0, 2, 3] 
 
plotter = pv.Plotter() 
plotter.add_mesh(unstructured, color=[0.6, 0.2, 0.1], label='beamy')
plotter.add_mesh(poly, color=[0.1, 0.6, 0.6], label='anty') 
 
plotter.add_legend() 
plotter.show()             

Как видите, строки передаются как label аргумент ключевого слова add_mesh превращаются в метки в легенде, которые можно включить с помощью add_legend()вызов. Параметры настройки легенды см. В документации.