向量化函数(Python)

问题描述

我是Python新手，正在尝试执行硬件分配，它不断向我抛回此错误(在IPython笔记本中)：

"TypeError: unsupported operand type(s) for ** or pow(): 'list' and 'float'"

这里是我试图模仿的公式：h(X)=(1/√2π)*e^−(1/2)^x^2

import numpy as np
import math
from math import *


def h_scalar(x):
    return (1 / sqrt(2 * pi)) * exp(-.50** x ** 2.00)

我正在努力取得和我老师一样的成绩。

l = [-10.0, -1.0, 0.0, 1.0, 10.0] #
print h_scalar(l)
>>> [1.4867195147342977e-06, 0.24197072451914337, 0.3989422804014327,
0.24197072451914337, 1.4867195147342977e-06]

这是老师们的问题btw：

首先，编写一个名为h_scalar的函数。该函数应该期待一个列表(或者通常是一个可迭代的项)。它将期望该列表包含浮点值。它应该返回一个新的浮点数列表，这些浮点数是输入列表中每个x值的h(X)。您应该将函数命名为h_scalar()

---

解决方案

使用numpy的sqrt和exp将获得更好的性能，它们将按元素运行：

import numpy as np
from numpy import sqrt, exp, pi
def h_scalar(x):
    x = np.array(x)
    return (1/sqrt(2*pi)) * exp(-0.50*x**2)

您可以这样使用：

x = [-10.0, -1.0, 0.0, 1.0, 10.0]
h_scalar(x)