在python中将数组的所有成员包装为pi

Question

我正在尝试将2D数组的所有成员包装成pi，但每当我告诉它打印时，我得到[nan，nan]。

这是我对pi功能的包装：

def wraptopi(x):
    pi = np.pi
    x = x - np.floor(x/(2*pi)) *2 *pi
    x[x >= pi] -= 2*pi

这是我尝试使用新数字填充数组：

y[0] = wraptopi(y[0]) 
y[1] = wraptopi(y[1])

所以在我的def wraptopi中代替x，我应该写一个函数吗？

Answer 1

看起来你不希望函数变异。以下是你应该改变的方式：

def wraptopi(x):
        pi = np.pi
        x = x - np.floor(x/(2*pi)) *2 *pi
        return x[x >= pi] - 2*pi

y[0] = wraptopi(y[0]) 
y[1] = wraptopi(y[1])