Numpy矩阵的映射函数

时间:2018-05-08 20:19:58

标签: python numpy matrix

我有两个numpy矩阵。一个包含lambda函数。另一个包含值。

是否有类似于Python的地图功能的功能可以让我获得预期的结果?

有更好的方法吗?

functionMatrix = np.array([[lambda x:x**2, lambda x:x**3],[lambda x: x**2, 
lambda x: np.sqrt(x)]])
valueMatrix = np.array([[1,2],[3,4]])

expectedResult = np.array([[1,8],[9,2]])

1 个答案:

答案 0 :(得分:3)

这只是语法糖,但可以胜任。

@np.vectorize
def apply_vec(f, x):
    return f(x)

result = apply_vec(functionMatrix, valueMatrix)