Python 函数适用于标量，但不适用于向量

Question

我是 python 新手，并且有一些 matlab 背景。我正在尝试为高斯核编写一个均值为 0 的函数。这是我的代码：import numpy as npimport mathdef GaussainKernel(x,sigma):    xx = np.array(x)    return 1/(np.sqrt(2*math.pi*sigma**2))*math.exp(-(xx*xx)/(2*sigma**2))如果传递 sigma 和 x 的标量，则此方法有效，例如：GaussainKernel(1, 1)给了我答案：0.24197072451914337现在，如果我想传递 x 的向量，例如：x = np.array([0,1,2])GaussainKernel(x, 1)我收到以下错误：TypeError: only size-1 arrays can be converted to Python scalars这必须是由于术语x*x或x**2任何帮助将不胜感激。

Answer 1

math.exp是一个Python函数，不接受向量作为输入。替换它np.exp，一切都会按预期工作。

另外，您可能需要替换math.pi为np.pi以摆脱一次导入。