我有一些数组:
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
但是如何以更 NumPy 的方式做到这一点？
我试过np.apply_along_axis :
res = np.apply_along_axis(np.kron, 1, array)
但是我没有成功。
最佳答案
我不知道它是否更“numpy”，但绝对更Pythonic的方法是使用reduce内置函数:
reduce(np.kron, array)
关于python - 对数组中的所有元素应用函数，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/54869546/