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