如何使用Drake可视化Pydrake IIWA操作站？代码报错解决方案

可运行代码示例 #

from pydrake.multibody.parsing import Parser
from pydrake.examples import ManipulationStation
from pydrake.geometry import StartMeshcat
from pydrake.visualization import AddDefaultVisualization
from pydrake.systems.framework import DiagramBuilder

# 初始化操作站
station = ManipulationStation()
# 如需加载额外模型，可保留以下解析器代码
# parser = Parser(station.get_mutable_multibody_plant(), station.get_mutable_scene_graph())

# 构建Diagram
builder = DiagramBuilder()
builder.AddSystem(station)
meshcat = StartMeshcat()

# 关键修改：直接传入操作站内部的plant实例
AddDefaultVisualization(
    builder=builder,
    meshcat=meshcat,
    plant=station.get_multibody_plant()
)

# 必须完成Diagram构建并发布上下文才能显示可视化
diagram = builder.Build()
context = diagram.CreateDefaultContext()
diagram.ForcedPublish(context)

错误原因与修复说明 #

• 原报错RuntimeError: DiagramBuilder does not contain a subsystem named plant：AddDefaultVisualization默认会在DiagramBuilder的顶层子系统中查找名为plant的组件，但ManipulationStation是封装好的Diagram，其内部的multibody_plant并非顶层子系统，因此找不到。
• 修复核心：通过plant参数直接传入station.get_multibody_plant()实例，绕过按名称查找的逻辑。
• 额外补充：原代码缺少Diagram构建和上下文发布步骤，这是可视化生效的必要操作。

内容的提问来源于stack exchange，提问作者wpy