NumPy的廣播機制可以讓不同維度的數組進行運算,而不需要手動擴展數組的維度。廣播機制的規則如下:
下面是一個簡單的例子,展示了如何使用廣播機制進行數組的運算:
import numpy as np
# 創建兩個數組
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 使用廣播機制進行數組運算
result = a + b
print(result)
在上面的例子中,數組a和數組b的維度不同,但是通過廣播機制,數組b被擴展為[4, 5, 6],然后和數組a進行元素級的加法運算。最終得到的結果是[5, 7, 9]。