NumPy的廣播功能可以讓不同形狀的數組進行數學運算,從而避免了手動擴展數組的麻煩。下面是一個簡單的示例,演示如何使用NumPy的廣播功能:
import numpy as np
# 創建一個數組
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
# 使用廣播將兩個數組相加
c = a + b
print(c)
在這個例子中,數組a
和數組b
的形狀相同,因此它們可以直接相加。但是如果數組形狀不同,NumPy會自動根據廣播的規則對數組進行擴展,使它們可以進行運算。例如:
import numpy as np
# 創建一個數組
a = np.array([[1, 2, 3], [4, 5, 6]])
b = np.array([10, 20, 30])
# 使用廣播將數組相加
c = a + b
print(c)
在這個例子中,數組a
的形狀是(2, 3)
,數組b
的形狀是(3,)
,但它們仍然可以進行相加運算,因為NumPy會自動將數組b
擴展為(2, 3)
的形狀再與數組a
相加。這樣,我們可以方便地對不同形狀的數組進行運算,而不必手動擴展數組的維度。