Created get_img function to save bike images

images/get_img.py

+import numpy as np
+import matplotlib.pyplot as plt
+from matplotlib.animation import FuncAnimation
+
+def save_bike_image(points_position, counter)
+    fig, ax = plt.subplots()
+    ax.set_title('Bikes')
+    ax.axis('equal')
+    ax.set_xlim(0, 35)
+    ax.set_ylim(-5, 20)
+
+    num_point=4
+    points_position = np.random.rand(num_point, 2)*8+3
+    wheel_radius = 1.2
+    spring_indexs = []
+    for i in range(num_point):
+        for j in range(num_point):
+            if i!=j:
+                spring_indexs.append([i,j])
+    spring_indexs=np.array(spring_indexs)
+    ax.plot(np.arange(-10, 40, 0.5), np.sin(np.arange(-10, 40, 0.5)), 'k')  # ground
+
+    balls = []
+    balls.append(plt.Circle(
+        (points_position[0][0], points_position[0][1]), wheel_radius, color='r'))
+    balls.append(plt.Circle(
+        (points_position[1][0], points_position[1][1]), wheel_radius))
+
+    lines = []
+    for item in spring_indexs:
+        lines.append(ax.plot([points_position[item[0], 0], points_position[item[1], 0]],
+                             [points_position[item[0], 1], points_position[item[1], 1]],
+                             'o-', markersize=9, markerfacecolor='b'))
+    for item in balls:
+        ax.add_patch(item)
+    plt.savefig(f"img_{counter}.png")