如何使用NumPy广播数组-开发者知识库平台

如何使用NumPy广播数组

numpy

1914

2024/5/29 13:27:52

栏目: 编程语言

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相加。这样，我们可以方便地对不同形状的数组进行运算，而不必手动扩展数组的维度。

如何使用NumPy广播数组