重复角度函数python

Question

  def ang(theta1,theta2,x1,x2,r,dens):
        theta = []
        r1 = []
        ds = []

       #note x1,x2 is input data

       theta = np.arctan2(x1,x2)
       ang_deg = (theta*180)/(np.pi)

       for i in range(len(ang_deg)):
            if theta1 <= ang_deg[i]<= theta2:
                r1.append(r[i])
                ds.append(dens[i])

       return 

此函数正常工作，但我想知道如何编码，而不是只说theta1 = 0和theta2 = 10度，我如何优化它以便可以多次执行例如从0,10，然后从10,20，然后从20,30度，分别给出r1和ds。

谢谢

Answer 1

您始终可以创建其他函数：

def ang_multiple(theta1s, theta2s, x1, x2, r, dens):
   results = []
   for i, theta1 in enumerate(theta1s):
       results.append(ang(theta1, theta2s[i], x1, x2, r dens))
   return results