import numpy as np

a = np.array([0,1,2])
b = np.array([3,4,5])
print(a+b)
print(a+5)
#
# 广播的规则
# 1. 如果两个数组维度个数不同,那么小维度的数组形状会在左边补1
# 2. 如果两个数组形状在任何一个维度上都不匹配,数组的形状会沿着维度为1的维度扩展已匹配另一个数组的形状
# 3. 如果没有维度形状=1,引发异常
a = np.arange(3)
b = np.arange(3).reshape(3,1)
print("a",a)
print("b\n",b)
print("a+b\n",a+b)
a = np.arange(6).reshape(2,3)
b = np.arange(6).reshape(3,2)
print("a",a)
print("b\n",b)
print("a+b\n",a+b)

numpy中的广播机制_数组

推导过程:

numpy中的广播机制_数组_02