Numpy数组的广播机制

Numpy数组的广播机制

当两个数组的形状并不相同的时候，我们可以通过扩展数组的方法来实现相加、相减、相乘等操作，这种机制叫做广播（broadcasting）

定义数据

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

c =np.array([[4],[5],[6]])

d = np.array([[1,2,3], [4,5,6]])

e = np.array([[1,2,3], [4,5,6],[7,8,9]])

1.1 一维数组与一维数组

a * b
# out：array([ 4, 10, 18])

1.2 一维数组和多维数组

a * c
# array([[ 4,  8, 12],
#        [ 5, 10, 15],
#        [ 6, 12, 18]])

1.3 多维数组和多维数组

c * e
# array([[ 4,  8, 12],
#        [20, 25, 30],
#        [42, 48, 54]])