Python numpy实现二维数组和一维数组拼接的方法

下面是详细讲解 "Python numpy实现二维数组和一维数组拼接的方法" 的攻略。

一、numpy.concatenate()方法

使用numpy的方法concatenate()可以实现二维数组和一维数组拼接。例如，我们有一个2x3的二维数组和一个大小为3的一维数组：

import numpy as np

a = np.array([[1, 2, 3],
              [4, 5, 6]])

b = np.array([7, 8, 9])

现在，我们想将b拼接到a的最后一行，可以使用concatenate()方法实现：

c = np.concatenate((a, np.array([b])))

在这里，我们使用了元组(a, np.array([b]))来表示我们要拼接的对象。c的值为：

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

可以看到，b被成功地拼接到了a的最后一行。

二、numpy.vstack()方法

另一个方便实用的numpy方法是vstack()，用于将一条序列vertically（沿着垂直方向）stack起来。它也可以用来拼接一维数组和二维数组。

例如，我们仍然有上面的 a 和 b 数组，我们可以使用vstack()方法来将 b 添加到 a 的末尾：

c = np.vstack((a, b))

这里的参数(a, b)是一个元组，它指定了要进行垂直堆叠的序列。输出如下：

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

三、示例说明

第一种情况

我们有一个大小为2x2的二维数组 a，和一个长度为2的一维数组 b，我们将 b 添加到 a 的最后一行：

a = np.array([[1, 2],
              [3, 4]])

b = np.array([5, 6])

c = np.concatenate((a, np.array([b])))

print(c)

输出结果为：

array([[1, 2],
       [3, 4],
       [5, 6]])

第二种情况

现在，我们有一个 2x2 的二维数组 a 和一个长度为 2 的一维数组 b，并且我们想要在 a 和 b 之间添加一行。我们可以使用 vstack() 方法来实现：

a = np.array([[1, 2],
              [3, 4]])

b = np.array([5, 6])

c = np.vstack((a, b))

print(c)

输出结果为：

array([[1, 2],
       [3, 4],
       [5, 6]])

这就是使用 numpy 实现二维数组和一维数组拼接的方法，希望能帮助到你。