Numpy 的數(shù)組運算與廣播

廣播，是指 Numpy 在算術(shù)運算期間處理不同形狀的數(shù)組的能力。

對數(shù)組的算術(shù)運算通常在相應(yīng)的元素上進行。 如果兩個陣列具有完全相同的形狀，則這些操作被無縫執(zhí)行，這一小節(jié)將通過數(shù)組的運算來展示廣播的一些特征。

1. 常見的兩種廣播途徑

1.1 相同大小的數(shù)組計算

如果兩個數(shù)組的維度完全一致，那么運算過程是兩個數(shù)組相應(yīng)元素的逐個計算。

案例

定義兩個完全一致的數(shù)組：

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

查看數(shù)組結(jié)構(gòu)：

arr0
out:
    array([[1, 2, 3],
		[4, 5, 6]])

arr1
out:
    array([[5, 5, 5],
       [5, 5, 5]])

查看加法運算的結(jié)果：

arr0 + arr1
out:
    array([[ 6,  7,  8],
       [ 9, 10, 11]])

查看乘法運算的結(jié)果：

arr0 * arr1
out:
    array([[ 5, 10, 15],
       [20, 25, 30]])

可以看到，如果數(shù)組的維度一致，那么廣播則是對應(yīng)位置的元素運算。

1.2 數(shù)組與標量的計算

廣播機制的存在，允許運算發(fā)生在不同維度的數(shù)組之間，其中最典型的場景，是數(shù)組的標量的計算。所謂的標量，我們可以簡單地將之視為是0維的數(shù)組。

案例

對于加法運算：

arr0 + 5
out:
    array([[ 6,  7,  8],
       [ 9, 10, 11]])

對于乘法運算

arr0 * 5
out:
    array([[ 5, 10, 15],
       [20, 25, 30]])

對于冪運算：

arr0 ** 5
out:
    array([[   1,   32,  243],
       [1024, 3125, 7776]], dtype=int32)

可以發(fā)現(xiàn)，對于上述效果，是數(shù)組每個元素分別與標量進行運算的結(jié)果。

直觀地，我們發(fā)現(xiàn)，標量在這個二維數(shù)組上發(fā)生了傳播（廣播），標量沿著數(shù)組的兩個維度擴散，直至擴散的結(jié)果和待計算的二維數(shù)組一致，最后進行計算。

2. 不同維度的數(shù)組運算

2.1 沿著一個維度進行廣播

一種簡單的場景是，兩個數(shù)組有一定的相似性，即數(shù)組（n×m）和數(shù)組（1,m）。

案例

arr2 = np.array([[10, 20, 30]])
arr2
out:
    array([[10, 20, 30]])

觀察不同大小的數(shù)組的廣播規(guī)則：

arr0 + arr2
out:
    array([[11, 22, 33],
       [14, 25, 36]])

arr0 * arr2
out:
    array([[ 10,  40,  90],
       [ 40, 100, 180]])

觀察發(fā)現(xiàn)，arr2沿著arr0的第二個維度擴展了，擴展到二者相匹配，再進行了對應(yīng)的計算。

可以用一張簡圖來進行描述相加的過程：

2.2沿著兩個維度同時廣播

涉及到兩個維度同時廣播，則相對復(fù)雜一些。

案例

arr3 = np.array([[10], [20], [30]])
arr3
out:
    array([[10],
       [20],
       [30]])

觀察沿著2個方向同時廣播的過程：

arr2 + arr3
out:
    array([[20, 30, 40],
       [30, 40, 50],
       [40, 50, 60]])

上述廣播的過程可以按照如下示意進行描述：

2.3 廣播的規(guī)則

所以根據(jù)上述案例，可以把廣播規(guī)則簡單總結(jié)如下：

• 讓所有輸入數(shù)組都向其中形狀最長的數(shù)組看齊，形狀中不足的部分都通過在前面加 1 補齊；
• 輸出數(shù)組的形狀是輸入數(shù)組形狀的各個維度上的最大值；
• 如果輸入數(shù)組的某個維度和輸出數(shù)組的對應(yīng)維度的長度相同或者其長度為 1 時，這個數(shù)組能夠用來計算，否則出錯；
• 當(dāng)輸入數(shù)組的某個維度的長度為 1 時，沿著此維度運算時都用此維度上的第一組值。

3. 小結(jié)

本節(jié)以數(shù)組的運算為基礎(chǔ)，介紹了數(shù)組在算術(shù)運算期間的廣播規(guī)則。在實際使用中，廣播的規(guī)則強大且靈活，當(dāng)條件不滿足時，則會拋出 ValueError: frames are not aligned 異常，需要注意。