Question

>>> import numpy as np

假设我有以下两个数组

>>> a = np.array([a1,a2,a3,a4,a5])
>>> d = np.array([d1,d2,d3])

我想知道如何操作这两个数组以获得5x3矩阵：

>>> m
   np.array([[a1*d1, a2*d1, a3*d1, a4*d1, a5*d1],
             [a1*d2, a2*d2, a3*d2, a4*d2, a5*d2],
             [a1*d3, a2*d3, a3*d3, a4*d3, a5*d3])
>>> m.shape
   (3, 5)

我无法找到合适的numpy功能来实现这一目标。即使最终矩阵出现了转换。

Answer 1

你需要使用广播：

a[None, :] * d[:, None]


In [2]: a = np.arange(5)

In [3]: d = np.arange(3)

In [4]: a[None, :] * d[:, None]
Out[4]: 
array([[0, 0, 0, 0, 0],
       [0, 1, 2, 3, 4],
       [0, 2, 4, 6, 8]])

您可以使用np.newaxis而不是None，它也是一样的（np.newaxis只是一个别名，名称更明确）。如何将其他轴添加到a或d：

In [5]: a[None, :].shape
Out[5]: (1, 5)

广播是一种非常有效的工具，可以在numpy中对数组执行操作，详细了解它in the doc。

作为评论，a * d[:, None]也可行，但我发现第一个表述更明确。

Answer 2

我喜欢@ P.Camilleri answer。这只是为了多样性......

>>> import numpy as np
>>> x,y = np.random.random(5), np.random.random(3)      
>>> x
array([0.5538738 , 0.31099761, 0.42709952, 0.28999541, 0.76811293])
>>> y
array([0.80281963, 0.31063054, 0.20141215])

>>> np.array([x])*np.array([y]).T
array([[0.44466076, 0.24967499, 0.34288388, 0.23281401, 0.61665614],
       [0.17205012, 0.09660536, 0.13267015, 0.09008143, 0.23859933],
       [0.11155691, 0.0626387 , 0.08602303, 0.0584086 , 0.15470727]])

Answer 3

一些时间：

>>> from timeit import timeit
>>> 
>>> kwds = dict(globals=globals(), number=10000)
>>> 
>>> a = np.random.random(100)
>>> d = np.random.random(60)
>>>
>>> timeit("d[:, None] * a", **kwds)
0.10106271505355835
>>> timeit("d[:, None] @ a[None]", **kwds)
0.08171644899994135
>>> timeit("np.dot(d[:, None], a[None])", **kwds)
0.08377223135903478
>>> timeit("d.reshape(-1, 1) * a", **kwds)
0.11084615904837847
>>> timeit("np.einsum('i,j->ij', d, a)", **kwds)
0.0618918058462441
>>> timeit("np.tensordot(d, a, ((),()))", **kwds)
0.23450013296678662
>>> timeit("np.c_[d] * a", **kwds)
0.19284361507743597
>>> timeit("np.multiply(*np.ix_(d, a))", **kwds)
0.16771024512127042
>>> timeit("np.multiply.outer(d, a)", **kwds)
0.09988314099609852
>>> timeit("np.outer(d, a)", **kwds)
0.11399263981729746

einsum看起来最快。

将1D阵列a [k]和d [j]乘以矩阵m [j，k]

3 个答案: