Question

在Python中，我有一个列表和一个numpy数组。

我想将数组乘以列表，以便得到一个数组，其中第三维代表输入数组乘以列表的每个元素。因此：

in_list = [2,4,6]
in_array = np.random.rand(5,5)
result = ...
np.shape(result) ---> (3,5,5)

其中（0，：，:)是输入数组乘以列表的第一个元素（2）; （1，：，:)是输入数组乘以list（4）的第二个元素等。

我有一种感觉，这个问题将通过广播来回答，但我不确定如何绕过这个问题。

Answer 1

你想要np.multiply.outer。 outer方法是为任何NumPy“ufunc”定义的，包括乘法。这是一个演示：

In [1]: import numpy as np

In [2]: in_list = [2, 4, 6]

In [3]: in_array = np.random.rand(5, 5)

In [4]: result = np.multiply.outer(in_list, in_array)

In [5]: result.shape
Out[5]: (3, 5, 5)

In [6]: (result[1, :, :] == in_list[1] * in_array).all()
Out[6]: True

正如您所建议的，广播提供了另一种解决方案：如果您将in_list转换为长度为3的1d NumPy数组，则可以重塑为形状(3, 1, 1)的数组，并且那么与in_array的乘法将适当地广播：

In [9]: result2 = np.array(in_list)[:, None, None] * in_array

In [10]: result2.shape
Out[10]: (3, 5, 5)

In [11]: (result2[1, :, :] == in_list[1] * in_array).all()
Out[11]: True

如何将numpy数组乘以列表以获得多维数组？

1 个答案: