关键词

详解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。

本文链接:http://task.lmcjl.com/news/17094.html

展开阅读全文