python中numpy.concatenate函数

numpy.concatenate函数

numpy.concatenate函数的主要作用：沿现有的某个轴对一系列数组进行拼接。

用法

import numpy as np

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

参数

• (a1, a2, ...)：输入的一系列数组，数组的形状必须相同，但与参数axis对应的维度（默认情况下是0轴）除外。
• axis：整数，表示数组的轴，通过指定该参数使其沿指定轴进行合并。
• out：

实例

实例1

拼接两个列表：

a = [2, 3, 4]
b = [1, 5, 2]

np.concatenate((a,b))

输出：

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

在实例1中，拼接的两个列表的形状均为(3,)，即，只有一个0轴，默认情况下，沿0轴进行拼接。

实例2

拼接两个列表：

a = [[2, 3, 4]]
b = [[1, 5, 2]]

np.concatenate((a,b))

输出：

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

在实例2中，拼接的两个列表的形状均为(1,3)，即，一个二维数组(0轴一个元素，1轴三个元素)，默认情况下，沿0轴进行拼接，输出(2,3)的数组。

指定沿1轴进行拼接

np.concatenate((a,b),axis=1)

输出：

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

沿1轴进行拼接，输出(1,6)的数组。