python 将一个 function 和两个 arguments 应用于矩阵的所有元素

Question

假设我有以下 function

def my_func(a,b):
    
    res = a[0] + a[1]*b
    
    return res

我知道如何将它应用于矩阵的一个元素：

import numpy as np
mydata = np.matrix([[1, 2], [3, 4]])
my_par = np.array([1, 2])
res = my_func(my_par,mydata[1,1])

我现在想将它应用于矩阵 mydata 的所有元素。 我已经尝试过

myfunc_vec = np.vectorize(my_func)
res = myfunc_vec(my_par,mydata)

我有以下错误：

in my_func
    res = a[0] + a[1]*b
IndexError: invalid index to scalar variable.

我认为错误是由于我将两个 arguments 传递给 function。

有什么方法可以将我的 function 应用于矩阵的所有元素而不会出错？

Answer 1

我认为最简单的方法是使用for循环。 确保也用np.matrix()替换 np.matrix np.array() 。

def my_func(a,b):  
    res = a[0] + a[1]*b  
    return res

import numpy as np
mydata = np.array([[1, 2], [3, 4]])
my_par = np.array([1, 2])
res = my_func(my_par,mydata[1,1])
res = np.zeros((len(mydata), len(mydata[0])))
for i in range(len(mydata)):
   for j in range(len(mydata[0])):
      res[i][j] = my_func(my_par, mydata[i][j])

print(res)

Output：

[[3. 5.]
 [7. 9.]]

希望有帮助！

Answer 2

您无需执行任何操作。 只需传递my_data而不是my_data[1,1]和 rest 一切都会到位。

mydata = np.matrix([[1, 2], [3, 4]])
my_par = np.array([1, 2])
res = my_func(my_par,mydata)
print(res)

Output：

[[3 5]
 [7 9]]