使用 Add 函数

介绍

在本教程中，我们将逐步讲解如何使用 NumPy 库中的 add() 函数。add() 函数可以将两个数组的元素进行拼接，但要求两个数组的形状必须相同。

前提条件

要学习本教程，你需要对 Python 和 NumPy 有基本的了解。

虚拟机使用提示

虚拟机启动完成后，点击左上角切换到 Notebook 标签页，以访问 Jupyter Notebook 进行练习。

有时，你可能需要等待几秒钟，直到 Jupyter Notebook 完成加载。由于 Jupyter Notebook 的限制，操作验证无法自动化。

如果你在学习过程中遇到问题，可以随时向 Labby 寻求帮助。请在课程结束后提供反馈，我们将及时为你解决问题。

导入 NumPy

第一步是导入 NumPy 库。

import numpy as np

创建输入数组

接下来，我们创建两个输入数组，用于测试 add() 函数。

x1 = ['Hello', 'World']
x2 = ['NumPy', 'Tutorial']

应用 add() 函数

要拼接两个输入数组的元素，我们可以使用 add() 函数，如下所示。

result = np.char.add(x1, x2)

打印结果

让我们打印上一步中得到的拼接后的字符串数组。

print(result)

输出结果为：

array(['HelloNumPy', 'WorldTutorial'], dtype='<U14')

对单元素数组应用 add() 函数

如果两个数组都只有一个元素，我们也可以应用 add() 函数。

x1 = ['Hello']
x2 = ['LabEx!']
result = np.char.add(x1, x2)
print(result)

输出结果为：

array(['HelloLabEx!'], dtype='<U18')

对不同长度数组应用 add() 函数

最后，让我们看看当输入数组长度不同时，add() 函数的行为。

x1 = ['Welcome', 'to', 'LabEx']
x2 = ['Best Place', 'Forlearning']
result = np.char.add(x1, x2)
print(result)

输出结果将是一个 ValueError：

ValueError: shape mismatch: objects cannot be broadcast to a single shape

这是因为两个数组的长度不同，无法进行拼接。

总结

在本教程中，我们学习了如何使用 NumPy 中的 add() 函数来拼接两个数组的元素。我们还了解到，为了使函数正常工作，两个数组的形状必须相同。