你可以在你的函数上使用np.vectorise来获得你的自定义乘法函数，使用所有常见的numpy特性，比如广播。

def my_multiplication(a, b):
    #your code that works on multiplying 2 numbers
    return c
v_my_multiplication = np.vectorize(my_multiplication)
v_my_multiplication([1, 2, 3], [1, 6])

#Will now work for np.array instead of just 2 numbers and utilize the broadcasting and vectorized implementation benefits that numpy has to offer.

尝试numpy.dot或x.dot(y)。请参阅文档here

• • 示例**

import numpy as np
x = np.arange(12).reshape((3,4))
y = np.arange(4)
print(x,"\n\n",y,"\n")
print (np.dot(x,y))

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]] 

 [0 1 2 3] 

[14 38 62]