numpy dstack 用法及代码示例

用法:
numpy.dstack(tup)
沿深度方向(沿第三轴)按顺序堆叠数组。

这等效于形状(M,N)的2-D数组已重塑为(M,N,1)和形状(N,)的1-D数组已重塑为(1, N,1)。重建除以的数组dsplit。

此功能对最多3维的阵列最有意义。例如,对于具有高度(第一轴),宽度(第二轴)和r /g /b通道(第三轴)的pixel-data。功能concatenate,stack和block提供更多常规堆栈和串联操作。

描述
参数 tup: : sequence of arrays

除第三个轴外,所有阵列的形状都必须相同。一维或二维阵列必须具有相同的形状。

返回值: stacked: : ndarray

通过堆叠给定阵列形成的阵列将至少为3D。

例子:

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

发表回复

登录... 后才能评论