将函数应用于数组中的所有元素

Question

我有一些数组：

array = np.array([[1, 0], [1, 0], [1, 0]])

我想以这种方式计算np.kron：

def func(array):
    res = array[0]
    for i in range(1, len(array)):
        res = np.kron(res, array[i]) 
    return res

但是如何以更麻木的方式做到呢？

我尝试了np.apply_along_axis：

res = np.apply_along_axis(np.kron, 1, array)

但是我没有成功。

Answer 1

我不知道它是否更“ numpy”，但是绝对更Python化的方式是使用reduce内置函数：

reduce(np.kron, array)