mp数组通过一个函数

Question

StackExchange的人，你好

我正在使用库mpmath作为项目，我需要一些数组帮助，即通过函数传递它们。

使用mpf元素创建数组没有问题。这段代码摘录让我这样做：

(...)

ds = 0.001

str_ds = str(ds); ds = mpf(str_ds)

s1 = arange(pi/mpf('2.0'), (3.0/2.0) * pi + ds/mpf('2.0'), ds)  
s2 = arange(pi/mpf('2.0'),-(1.0/2.0) * pi - ds/mpf('2.0'),-ds)

s1 = array(s1)
s2 = array(s2)

def r1(s):
    return array([xc + r * cos(s), r * sin(s)])

def r2(s):
    return array([xc + L + r * cos(s), r * sin(s)])

但是，如果我打电话到下一行：

r1 = r1(s1)

r2 = r2(s2)

它给了我一个错误。有没有办法通过函数传递一个带有mpf元素的数组？

编辑：我尝试在每个r1和r2函数中使用vectorize，如下所示：

cos = vectorize(cos)
sin = vectorize(sin)

但它一直没有用。