如何将numpy二维数组与numpy一维数组相乘？

Question

两个数组：

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

我要的是：

c = [[6,9,6], 
    [25,30,5]] 

但是，我得到这个错误：

ValueError: operands could not be broadcast together with shapes (2,3) (2) 

如何乘钕阵列1D数组，其中len(1D-array) == len(nD array)？

Answer 1

你需要数组转换B到A（2，1）形状数组，使用没有或numpy.newaxis在索引元组：

import numpy 
a = numpy.array([[2,3,2],[5,6,1]]) 
b = numpy.array([3,5]) 
c = a * b[:, None] 

这里是document。

Answer 2

另一种策略是reshape所述 第二阵列，因此它具有相同的维数与第一阵列：

c = a * b.reshape((b.size, 1)) 
print(c) 
# [[ 6 9 6] 
# [25 30 5]] 

可替代地，第二阵列的shape属性可以就地进行修改：

b.shape = (b.size, 1) 
print(a.shape) # (2, 3) 
print(b.shape) # (2, 1) 
print(a * b) 
# [[ 6 9 6] 
# [25 30 5]]