Python 類 __call__ 函數未矢量化

Question

我正在嘗試使用__call__ dunder 為自定義類繪制函數

import math
import numpy as np
import matplotlib.pyplot as plt

class F:
    def __init__(self, n, m):
        self._n = n
        self._m = m

    def __call__(self, x):
        return math.sin(self._n * x) * math.cos(self._m * x)

u = F(1,1)
v = F(1,1)

x_array = np.linspace(0, 2*np.pi, 20)

u_array = u(x_array)
v_array = v(x_array)

plt.plot(u_array, v_array)
plt.show()

但是，執行代碼時，我收到此錯誤

Traceback (most recent call last):
  File "F.py", line 18, in <module>
    u_array = u(x_array)
  File "F.py", line 11, in __call__
    return math.sin(self._n * x) * math.cos(self._m * x)
TypeError: only size-1 arrays can be converted to Python scalars

似乎調用u_array = u(x_array)將整個 x_array 發送到函數，而不是以矢量化的方式進行。

請幫忙

Answer 1

使用numpy向量化函數而不是來自math函數：

def __call__(self, x):
    return np.sin(self._n * x) * np.cos(self._m * x)