np.concatenate 用法详解

概述

np.concatenate 是NumPy库的一个重要函数，用于在NumPy数组中进行数组合并。它可以将两个或多个一维或更高维的数组在指定轴上连接起来，形成新的数组。本文将详细介绍np.concatenate函数的用法，并通过实际案例进行演示。

函数定义

np.concatenate((a1, a2, ...), axis=0)

参数：

• a1, a2, …：要连接的数组，可以为一维或更高维数组。
• axis：指定连接轴，默认为0。

返回值：

• 返回连接后的数组。

使用方法

连接一维数组

import numpy as np

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

result = np.concatenate((a1, a2))
print(result)

输出：

[1 2 3 4 5 6]

连接二维数组

a1 = np.array([[1, 2], [3, 4]])
a2 = np.array([[5, 6], [7, 8]])

result = np.concatenate((a1, a2), axis=1)
print(result)

输出：

[[1 2 5 6]
 [3 4 7 8]]

连接三维数组

a1 = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
a2 = np.array([[[13, 14, 15], [16, 17, 18]], [[19, 20, 21], [22, 23, 24]]])

result = np.concatenate((a1, a2), axis=1)
print(result)

输出：

[[[ 1  2  3]
  [ 4  5  6]
  [13 14 15]
  [16 17 18]]

 [[ 7  8  9]
  [10 11 12]
  [19 20 21]
  [22 23 24]]]

总结

np.concatenate 函数是 NumPy 库中一个非常实用的数组合并函数，可以用于连接一维或更高维的数组。通过指定连接轴，可以实现灵活的数组合并操作。在实际编程中，np.concatenate函数可以大大提高数组处理的效率。