Python numpy 数组减法

在Python中，使用NumPy的数组减法操作可以快速的对数组进行减法运算。下面让我们一起来看看如何使用NumPy进行数组减法。

安装NumPy

如果你还没有安装NumPy，你可以在终端下使用以下命令进行安装：

pip install numpy

通过NumPy进行数组减法

NumPy库提供了非常灵活的数组减法操作。我们可以简单地将两个数组进行减法运算，得到一个新的数组。

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([5, 4, 3, 2, 1])
result = arr1 - arr2

print(result)

这个程序会输出如下结果：

[-4, -2, 0, 2, 4]

在本例中，我们创建了两个NumPy数组 arr1 和 arr2，分别包含了一些数字。然后将它们相减，并将结果存储在 result 变量中。

我们还可以对不同形状的数组进行减法运算。如果两个数组的形状不同，则需要根据Broadcasting规则进行扩展，以便它们能够进行减法运算。

arr1 = np.array([[1, 2], [3, 4], [5, 6]])
arr2 = np.array([1, 2])
result = arr1 - arr2

print(result)

这个程序会输出如下结果：

[[0, 0],
 [2, 2],
 [4, 4]]

在本例中，我们创建了一个二维数组 arr1 和一个一维数组 arr2。我们将它们相减，由于 arr2 的形状被扩展为了 (2,) 的形状，而 arr1 的形状是 (3, 2)，所以 arr2 被复制了三次，使得两个数组的形状匹配，从而进行减法运算。

总结

以上介绍了如何使用Python NumPy库进行数组减法操作。无论您是在做简单的数组减法运算，还是需要进行Broadcasting扩展后的数组运算，NumPy都可以帮助您快速方便地进行计算。