python - 関数による 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 関数内でベクトル化を使用してみました:

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

しかし、それは機能しません。