numpy.dstack详解

最新推荐文章于 2025-02-14 15:15:19 发布

zhenyu wu

最新推荐文章于 2025-02-14 15:15:19 发布

阅读量2.3w

点赞数 13

分类专栏： numpy 文章标签： python numpy 机器学习人工智能线性代数

本文链接：https://blog.csdn.net/wzy628810/article/details/103840261

版权

numpy.dstack函数用于将二维或一维数组在深度方向（第三轴）上拼接。适用于处理3维数组，如图像数据的RGB通道。它接受一个元组作为输入，并在第三轴上连接数组。举例来说，将两个2D数组拼接成3D数组。了解更多详情，可参考提供的github链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

numpy.dstack

用例:
numpy.dstack(tup)
功能:
将列表中的数组沿深度方向进行拼接。
当数组为2维数组(M,N)或1维数组(N,)时，首先分别将其维度改变为(M,N,1)、(1,N,1)，然后沿着第三根轴进行拼接。使用函数dsplit可以将数组沿深度方向进行分隔。
这个函数适用于3维数组，例如图像数据有高（第一根轴），宽（第二根轴），和r/g/b通道（第三根轴）。concatenate、stack和block这三个函数为数组的堆叠以及拼接提供了更多选项。
让我们先观察一下数组拼接的三个方向:

# 绘制多维数组拼接以及分割的三个轴方向，first axis为竖直方向，second axis为水平方向，third axis为深度方向
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure() 
ax = plt.axes(projection='3d')
x = [0, 0