Numpy `ndarray` 数组沿第二轴连接两个一维数组

在许多情况下，我们需要合并两个相同类型和相同维度的数组。例如，在 numpy 中，我们需要将两个一维数组沿第二轴（列）拼接在一起，以创建一个新的二维数组。numpy 的 concatenate() 函数可以完成这个操作。在这篇文章中，我们将介绍如何用 numpy 将两个一维数组沿第二轴连接起来。

阅读更多：Numpy 教程

创建一维数组

首先，我们需要创建两个一维数组。我们可以使用 numpy 中的 array() 函数来创建一维数组。以下是一个例子：

import numpy as np

# 创建两个一维数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

现在，我们有两个一维数组 arr1 和 arr2。我们可以使用 print() 函数来打印这些数组，以检查它们的内容。以下是一个例子：

print("数组 1：")
print(arr1)
print("\n数组 2：")
print(arr2)

运行此代码将输出以下内容：

数组 1：
[1 2 3]

数组 2：
[4 5 6]

这表明我们已成功创建了两个一维数组。

沿第二轴连接两个一维数组

现在，我们想将这两个数组沿第二轴连接起来，以创建一个新的二维数组。我们可以使用 numpy 的 concatenate() 函数来执行此操作。以下是一个例子：

# 沿第二轴拼接两个数组
new_arr = np.concatenate((arr1, arr2), axis=1)

在这个例子中，我们传递了两个一维数组 arr1 和 arr2，并使用 axis=1 参数来指定我们要沿第二轴合并这些数组。函数将返回一个新的一维数组。

检查新的二维数组

现在，我们想检查我们是否成功创建了新的二维数组。我们可以使用 print() 函数来打印这个数组的内容。以下是一个例子：

print("合并后的数组：")
print(new_arr)

运行此代码将输出以下内容：

合并后的数组：
[1 2 3 4 5 6]

这表明我们已经成功地将两个一维数组沿第二轴连接起来，创建了一个新的二维数组。

将两个一维数组沿第一轴连接

除了沿第二轴连接两个一维数组外，我们还可以沿第一轴连接两个一维数组。以下是一个例子：

import numpy as np

# 创建两个一维数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

# 沿第一轴拼接两个数组
new_arr = np.concatenate((arr1, arr2), axis=0)

# 打印合并后的数组
print(new_arr)

在此示例中，我们传递了两个一维数组 arr1 和 arr2，并使用默认的 axis=0 参数来指定我们要沿第一轴合并这些数组。函数将返回一个新的一维数组。