train running

lib/objects/trajectory.py

@@ -41,6 +41,12 @@ class Trajectory(object):
     def as_paired_list(self):
         return list(zip(self._vertices[:-1], self._vertices[1:]))
 
+    def draw_in_array(self, shape):
+        trajectory_space = np.zeros(shape)
+        for index in self.vertices:
+            trajectory_space[index] = 1
+        return trajectory_space
+
     @property
     def np_vertices(self):
         return [np.array(vertice) for vertice in self._vertices]