多个numpy 2d数组的元素乘法

如何解决多个numpy 2d数组的元素乘法

为了简化我的问题，假设我有这些数组：

a = np.array([[1,2,3],[4,5,6]])
b = np.array([[2,2],[3,3,3]])
c = np.array([[1,1,6]])

我想对它们使用逐元素乘法，所以结果是：

array([[  2,4,18],[ 48,15,108]])

我知道我可以做 a*b*c，但是如果我有很多 2d 数组或者我不知道数组的数量，这将不起作用。我也知道 numpy.multiply 但这仅适用于 2 个数组。

解决方法

使用 stack 和 prod。

stack 将创建一个可以沿轴减少 prod 的数组。

import numpy as np

a = np.array([[1,2,3],[4,5,6]])
b = np.array([[2,2],[3,3,3]])
c = np.array([[1,1,6]])

unknown_length_list_of_arrays = [a,b,c]

d1 = a * b * c
stacked = np.stack(unknown_length_list_of_arrays)
d2 = np.prod(stacked,axis=0)

np.testing.assert_equal(d1,d2)