Simulate robot arm using python

import math
import matplotlib.pyplot as plt

l1 = 0.8
l2 = 0.5

n_theta = 10 # No of divisions
theta_start = 0 # Starting angle
theta_end = math.pi/2 # Ending angle
theta1 = []
theta2 = []

for i in range(0,n_theta):
   theta1.append(theta_start+ (theta_end-theta_start)*i/(n_theta-1)) # Angles of link 1
   theta2.append(theta_start+ (theta_end-theta_start)*i/(n_theta-1)) # Angles of link 2

# Base posotion 
x0 = 0 # Base position
y0 = 0 # Base position
ct = 1 # Counter

# Link 1 end point
for t1 in theta1:
	
	x1 = l1*math.cos(t1) # End of link 1
	y1 = l1*math.sin(t1) # End of link 1
	for t2 in theta2:
		
		x2 = x1+l2*math.cos(t2) # End of link 2
		y2 = y1+l2*math.sin(t2) # End of link 2

		filename = str(ct) + '.png'
		ct = ct+1
		plt.figure()
		plt.plot([x0,x1],[y0,y1])
		plt.plot([x1,x2],[y1,y2])
		plt.xlim([-1,2])
		plt.ylim([-1,2])
		plt.savefig(filename)

Source: skill-lync.com

Similar codes

9 votes221 views4 answers
Convert float to integer pandas
Python
4 votes196 views1 answer
How to convert types of variablesin python
Python
0 vote206 views1 answer
Say something in discord discord.py
Python
0 vote232 views1 answer
Python read from stdin pipe
Python
0 vote233 views1 answer
Datafram combine 3 columns to datetime
Python
0 vote214 views1 answer
String without space pythonm
Python
4 votes211 views2 answers
Python list transpose
Python
0 vote228 views1 answer
Compute confusion matrix using python
Python
0 vote219 views2 answers
Unpacking tuples in python
Python
0 vote219 views2 answers
Fix ImportError: No module named PIL
Python