详解Numpy dstack()（深度堆叠数组）函数的作用与使用方法

Numpy dstack()是Numpy库中的一个函数，用于沿深度方向将数组进行堆叠，具体来说，它将相同尺寸的数组沿着第三个维度（深度方向）进行水平叠加，返回一个新的数组。其语法如下：

numpy.dstack(tup)

其中，tup是一个由要堆叠的数组组成的序列。

下面给出两个使用dstack()函数的实例来更好地理解它的作用和用法：

将两个二维数组进行堆叠

import numpy as np

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

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

c = np.dstack((a, b))   # 在第三维度上堆叠a和b数组
print(c)

输出结果为：

Out: 
array([[[1, 5],
        [2, 6]],

       [[3, 7],
        [4, 8]]])

可以看到，dstack()函数将a和b数组在深度方向进行堆叠，生成了一个新的3维数组c。

将三个一维数组进行堆叠

import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.array([7, 8, 9])

d = np.dstack((a, b, c))    # 在第三维度上堆叠a、b和c数组
print(d)

输出结果为：

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

可以看到，dstack()函数将一维数组a、b和c在深度方向进行堆叠，生成了一个新的3维数组d。