如何使用NumPy广播数组

   2024-10-20 7010
核心提示:NumPy的广播功能可以让不同形状的数组进行数学运算,从而避免了手动扩展数组的麻烦。下面是一个简单的示例,演示如何使用NumPy的

NumPy的广播功能可以让不同形状的数组进行数学运算,从而避免了手动扩展数组的麻烦。下面是一个简单的示例,演示如何使用NumPy的广播功能:

import numpy as np# 创建一个数组a = np.array([1, 2, 3])b = np.array([4, 5, 6])# 使用广播将两个数组相加c = a + bprint(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 + bprint(c)

在这个例子中,数组a的形状是(2, 3),数组b的形状是(3,),但它们仍然可以进行相加运算,因为NumPy会自动将数组b扩展为(2, 3)的形状再与数组a相加。这样,我们可以方便地对不同形状的数组进行运算,而不必手动扩展数组的维度。

 
举报打赏
 
更多>同类网点查询
推荐图文
推荐网点查询
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号