在Numpy中正确展开反正切函数

Question

我想找出一个更好的方法来展开numpy的arctan函数的输出。假设我有：

import numpy as np
pi = np.pi

angles = np.deg2rad(range(0,5*360))
tangent = np.tan(angles)
arctangent = np.arctan(tangent)

现在我得到了它们返回的角度是从-pi/2到{}，但是我希望它们返回展开的（从0到{}）。使用numpy函数np.unwrap不适用于此，我不确定原因，所以我一直在使用我自己的函数：

^{pr2}$

它会返回原始角度。我在想办法解决这个问题np.展开自己去做，但是搞不清楚。有人知道怎么做吗？在

Answer 1

由于np.tan和np.arctan都返回一个数组，所以即使输入是一个列表，unwrap也可以写成：

np.unwrap(2*(x+np.pi/2))/2-np.pi/2

对于测试值

^{pr2}$

有效。但你大概知道你在添加pi/2时在做什么。在

np.unwrap是纯numpy Python，因此您可以轻松地研究其方法。在