Python NumPy dstack方法用法及代码示例

Numpy 的 dstack(~) 方法用于沿第三轴连接数组。检查下面的示例以进行说明。

参数

1. tup | array-like

将沿第三轴连接的数组。

返回值

一个 Numpy 数组，其中提供的数组沿第三轴串联。

例子

连接一维数组

x = np.array([1,2,3]) 
y = np.array([4,5,6]) 
z = np.dstack([x,y])
z



array([[[1, 4],
        [2, 5],
        [3, 6]]])

请注意，您不限于一次仅连接两个数组 - 您可以传入任意数量的数组。另外，这里返回的 Numpy 数组是三维的。

连接二维数组

x = np.array([[1,2],[3,4]]) 
y = np.array([[5,6],[7,8]]) 
z = np.dstack([x,y])
z



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

为了更容易处理这个问题，这里是x和y pretty-formatted：

x = [[1, 2],     y = [[5, 6] 
     [3, 4]]          [7, 8]]

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | dstack method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。