#### 廣播機制
在Numpy中當(dāng)數(shù)組進行運算時,如果兩個數(shù)組的形狀相同,那么兩個數(shù)組相加就是兩個數(shù)組的對應(yīng)位相加,這是要求維數(shù)相加,并且各維度的長度相同。比如:
```
import numpy as np
data1 = np.arange(9,dtype=np.int32).reshape(3,3) # 維數(shù)是(3,3)
data2 = np.ones((3,3),dtype=np.int32) # 維數(shù)是(3,3)
# 兩個相加
print(data1+data2)
```
原來兩個數(shù)是:
```
[[0 1 2]
[3 4 5]
[6 7 8]]
[[1 1 1]
[1 1 1]
[1 1 1]]
```
相加之后的結(jié)果:
```
[[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
```
當(dāng)運算中兩個數(shù)組的形狀不同使時,numpy將會自動觸發(fā)廣播機制,那什么是廣播機制呢?
復(fù)習(xí)下數(shù)學(xué)知識,在線性代數(shù)中我們曾經(jīng)學(xué)到過如下規(guī)則:
a1 =3 ,a2 = 4,a1,a2是0維張量,即標(biāo)量;b1,b2是1維張量,即向量; c1,c2是如下所示的2維張量,即矩陣:
a1與a2之間可以進行加減乘除,b1與b2可以進行逐元素的加減乘除運算,c1與c2之間可以進行逐元素的加減乘除以及矩陣相乘運算(矩陣相乘必須滿足維度的對應(yīng)關(guān)系),而a與b,或者b與c之間不能進行逐元素的加減乘除運算,原因是他們的維度不匹配。而這種在數(shù)學(xué)方面的不可能在NumPy中,就可以通過廣播完成這項操作。
再比如:
```
import numpy as np
data1 = np.arange(9,dtype=np.int32).reshape(3,3) # 維數(shù)是(3,3)
print(data1+1)
```
此時data1是3行3列的矩陣,跟一個1進行運算,能否成功呢?在Numpy中這時ok的。data1中的每個元素都會跟1相加而得到一個新的矩陣。這就是廣播機制。所以結(jié)果就是:
```
[[1 2 3]
[4 5 6]
[7 8 9]]
```
如果是跟一個3行1列的進行加法呢?
```
import numpy as np
data1 = np.arange(9,dtype=np.int32).reshape(3,3) # 維數(shù)是(3,3)
data2 = np.array([[1],[2],[3]])
print(data1+data2)
```
這個操作也是ok的,結(jié)果是:
```
[[ 1 2 3]
[ 5 6 7]
[ 9 10 11]]
```
如果是跟一個2行3列的數(shù)據(jù)進行加法運算呢?
```
import numpy as np
data1 = np.arange(9,dtype=np.int32).reshape(3,3) # 維數(shù)是(3,3)
data2 = np.array([[1,2,3],[1,1,1]])
print(data1+data2)
```
此時會報錯:
```
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
in
----> 1 print(data1+data2)
ValueError: operands could not be broadcast together with shapes (3,3) (2,3)
```
報錯的原因是什么呢?我們一起來看一張圖
所以廣播的規(guī)則是:
1. 形狀相同的廣播
```
import numpy as np
data1 = np.arange(9,dtype=np.int32).reshape(3,3)
print(data1+data1)
```
2. 相同維度,但其中某一個或多個維度長度為 1 的廣播:
```
import numpy as np
data1 = np.arange(9,dtype=np.int32).reshape(3,3) # 維數(shù)是(3,3)
data2 = np.array([[1],[2],[3]])
print(data1+data2)
data2 = data2.T
print(data1+data2)
```
3. 如果是標(biāo)量的話,會廣播整個數(shù)組上
```
import numpy as np
data1 = np.arange(9,dtype=np.int32).reshape(3,3) # 維數(shù)是(3,3)
print(data1+5)
```
所以我們要首先了解numpy的廣播機制,接下來才能更好的進行數(shù)組的運算。
更多關(guān)于“Python培訓(xùn)”的問題,歡迎咨詢千鋒教育在線名師。千鋒教育多年辦學(xué),課程大綱緊跟企業(yè)需求,更科學(xué)更嚴(yán)謹,每年培養(yǎng)泛IT人才近2萬人。不論你是零基礎(chǔ)還是想提升,都可以找到適合的班型,千鋒教育隨時歡迎你來試聽。
注:本文部分文字和圖片來源于網(wǎng)絡(luò),如有侵權(quán),請聯(lián)系刪除。版權(quán)歸原作者所有!