numpy的广播

2018-08-25  本文已影响24人  markDownMan

参考:numpy中的广播机制 - 小舔哥 - 博客园


把数组铺开,扩展之后无法匹配就无法进行广播。

可以广播的条件:

1.数组维度不同,轴长度相同

a =np.array( [ [1,2,3] , [3,2,1] ] )

b= np.array( [ [1,1,1] ] )

print(a+b)

output:

[[2 3 4]

[4 3 2]]


2.数组维度相同,且有个轴长度为1或者0

a =np.array( [ [1,2,3] , [3,2,1] ] )

b= np.array( [ [2] , [3] ] )

print(a+b)

ouput:

[[3 4 5]

[6 5 4]]

上一篇 下一篇

猜你喜欢

热点阅读