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]])