python - 如何将 numpy 元组数组乘以标量数组

Question

我有一个形状为 (N,2) 的 numpy 数组 A 和一个形状为 (N) 的 numpy 数组 S。

如何将两个数组相乘？目前我正在使用这段代码:

tupleS = numpy.zeros( (N , 2) )
tupleS[:,0] = S
tupleS[:,1] = S
product = A * tupleS

我是一名 Python 初学者。有更好的方法吗？

Answer 1

Numpy 使用行优先顺序，因此您必须显式创建一个列。如:

>> A = numpy.array(range(10)).reshape(5, 2)
>>> B = numpy.array(range(5))
>>> B
array([0, 1, 2, 3, 4])
>>> A * B
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: shape mismatch: objects cannot be broadcast to a single shape
>>> B = B.reshape(5, 1)
>>> B
array([[0],
       [1],
       [2],
       [3],
       [4]])
>>> A * B
array([[ 0,  0],
       [ 2,  3],
       [ 8, 10],
       [18, 21],
       [32, 36]])