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