如何使用Numpy库来合并两个数组|极客教程

如何使用Numpy库来合并两个数组

在数据处理和科学计算中，经常需要对数组进行操作，比如合并、拆分等。Numpy是Python中一个非常重要的库，它提供了大量的数组操作功能。本文将详细介绍如何使用Numpy库来合并两个数组，即使用numpy.append()函数。

1. numpy.append() 函数简介

numpy.append() 函数用于将两个数组合并成一个数组。该函数的基本语法如下：

numpy.append(arr, values, axis=None)

arr：指定要合并的第一个数组。
values：指定要合并的第二个数组。
axis：指定合并的轴。如果不指定，则结果是扁平化的。

2. 在无指定轴的情况下使用numpy.append()

当axis参数未指定时，numpy.append()函数会将所有输入数组扁平化后合并成一个一维数组。

示例代码1：合并两个一维数组

import numpy as np

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

Output:

如何使用Numpy库来合并两个数组

示例代码2：合并两个二维数组

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
result = np.append(arr1, arr2)
print(result)

Output:

如何使用Numpy库来合并两个数组

3. 在指定轴上使用numpy.append()

当axis参数被指定时，numpy.append()函数会沿着指定的轴合并数组。数组的形状在合并轴上除外，其他轴的形状必须相同。

示例代码3：沿着轴0合并两个二维数组

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
result = np.append(arr1, arr2, axis=0)
print(result)

Output:

如何使用Numpy库来合并两个数组

示例代码4：沿着轴1合并两个二维数组

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
result = np.append(arr1, arr2, axis=1)
print(result)

Output:

如何使用Numpy库来合并两个数组

4. 错误处理

在使用numpy.append()时，如果指定的轴的维度不匹配，则会抛出ValueError异常。

示例代码5：尝试沿不匹配的轴合并数组

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([5, 6])
try:
    result = np.append(arr1, arr2, axis=1)
    print(result)
except ValueError as e:
    print(e)

Output:

如何使用Numpy库来合并两个数组

5. 更多示例

示例代码6：合并多个一维数组

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
arr3 = np.array([7, 8, 9])
result = np.append(np.append(arr1, arr2), arr3)
print(result)

Output:

如何使用Numpy库来合并两个数组

示例代码7：合并多个二维数组

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
arr3 = np.array([[9, 10], [11, 12]])
result = np.append(np.append(arr1, arr2, axis=0), arr3, axis=0)
print(result)

Output:

如何使用Numpy库来合并两个数组

示例代码8：合并具有不同形状的数组

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([[4, 5, 6], [7, 8, 9]])
result = np.append(arr1, arr2)
print(result)

Output:

如何使用Numpy库来合并两个数组

示例代码9：使用flatten之后再append

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
arr1_flat = arr1.flatten()
arr2_flat = arr2.flatten()
result = np.append(arr1_flat, arr2_flat)
print(result)

Output:

如何使用Numpy库来合并两个数组

示例代码10：合并三维数组

import numpy as np

arr1 = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
arr2 = np.array([[[9, 10], [11, 12]], [[13, 14], [15, 16]]])
result = np.append(arr1, arr2, axis=0)
print(result)