不同尺寸的NumPy数组之间的减法。_编程开发

创始人

2025-01-08 10:01:27

0次

要进行不同尺寸的NumPy数组之间的减法，需要使用NumPy的广播（broadcasting）功能。广播是一种NumPy的机制，用于在不同形状的数组之间执行算术运算。

下面是一个示例代码，展示了如何使用广播进行不同尺寸的数组之间的减法：

import numpy as np

# 创建两个不同尺寸的数组
a = np.array([1, 2, 3])
b = np.array([[1, 2, 3], [4, 5, 6]])

# 使用广播进行减法
result = b - a

print(result)

输出：

[[0 0 0]
 [3 3 3]]

在这个示例中，数组a的形状是(3,)，数组b的形状是(2, 3)。通过广播，NumPy会自动将a扩展为(2, 3)的形状，使得两个数组可以进行减法运算。减法的操作是对应元素之间的减法，即b[0,0] - a[0]，b[0,1] - a[1]，以此类推。

需要注意的是，广播的规则是，对于每个维度，如果数组的形状在该维度上相等，或者其中一个数组在该维度上形状为1，那么这两个数组在该维度上是兼容的。如果两个数组的形状在某个维度上都不相等且都不为1，则广播操作会引发错误。

希望这个示例能够帮助你理解如何在NumPy中进行不同尺寸的数组之间的减法。