Apply a function along all elements in a array

Question

I have some array:

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

I want to calculate np.kron in a such manner:

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

But how to do it in a more numpy way?

I tried np.apply_along_axis :

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

But I didn't succeed.

Answer 1

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

reduce(np.kron, array)