Python 用于计算和绘制arctanx的函数

我需要写一个函数，它可以为我作为两个数组输入的值编写和求解tan的倒数。有时分母上的数组包含零，因此总是会出现被零除的情况。我不知道如何消除错误，让答案返回pi/2

def deviation(x):

if capture(x)==True:
 for i in range(len(yvf)):
          theta=sp.arctan(xvf/yvf)   #xvf and yvf are two arrays

---

首先：写入xvf[i]和yvf[i]以逐个访问阵列的单个元素

其次，识别yff[i]是否等于零：

if yvf[i] == 0:
    theta = math.pi/2.0
else:
    theta = sp.arctan(xvf[i]/yvf[i])

---

首先：写入xvf[i]和yvf[i]以逐个访问阵列的单个元素

其次，识别yff[i]是否等于零：

if yvf[i] == 0:
    theta = math.pi/2.0
else:
    theta = sp.arctan(xvf[i]/yvf[i])

---

如果导入numpy，我建议使用arctan2而不是arctan（）。此函数用于管理第二个参数中的零值

如果没有，可以使用math library的atan2函数和zip内置函数解决此问题：

import math

xvf = [0.,2.,2.]
yvf = [20.,0.,2.]

def arctan(xvf,yvf):
  return [math.atan2(x,y) for x,y in zip(xvf,yvf)]

print arctan(xvf,yvf)

---

如果导入numpy，我建议使用arctan2而不是arctan（）。此函数用于管理第二个参数中的零值

如果没有，可以使用math library的atan2函数和zip内置函数解决此问题：

import math

xvf = [0.,2.,2.]
yvf = [20.,0.,2.]

def arctan(xvf,yvf):
  return [math.atan2(x,y) for x,y in zip(xvf,yvf)]

print arctan(xvf,yvf)