溫馨提示×

如何使用NumPy廣播數(shù)組

小億
83
2024-05-06 13:27:52
欄目: 編程語言

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ù)組的維度。

0