From 2afa10a457769ed0eed92b5bfd47087492c860e1 Mon Sep 17 00:00:00 2001 From: karaage Date: Sun, 3 Jan 2021 13:50:54 +0900 Subject: [PATCH] Add plot initialize for google colab visualization (#453) --- ArmNavigation/n_joint_arm_3d/NLinkArm3d.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ArmNavigation/n_joint_arm_3d/NLinkArm3d.py b/ArmNavigation/n_joint_arm_3d/NLinkArm3d.py index b411cd71..3f6c5ddd 100644 --- a/ArmNavigation/n_joint_arm_3d/NLinkArm3d.py +++ b/ArmNavigation/n_joint_arm_3d/NLinkArm3d.py @@ -43,8 +43,6 @@ class Link: class NLinkArm: def __init__(self, dh_params_list): - self.fig = plt.figure() - self.ax = Axes3D(self.fig) self.link_list = [] for i in range(len(dh_params_list)): self.link_list.append(Link(dh_params_list[i])) @@ -64,6 +62,9 @@ class NLinkArm: alpha, beta, gamma = self.euler_angle() if plot: + self.fig = plt.figure() + self.ax = Axes3D(self.fig) + x_list = [] y_list = [] z_list = []