numpy.unstack#

numpy.unstack(x, /, *, axis=0)[source]#

沿着给定轴将数组拆分成一系列数组。

axis 参数指定将沿其拆分数组的维度。例如，如果 axis=0（默认值），它将是第一维，如果 axis=-1，它将是最后一维。

结果是沿着 axis 拆分的数组的元组。

2.1.0 版的新功能。

参数：
xndarray

要拆分的数组。

axisint, 可选

将沿其拆分数组的轴。默认值：0。

返回：
unstackedndarrays 的元组

拆分的数组。

参见

stack

沿着新轴连接一系列数组。

concatenate

沿着现有轴连接一系列数组。

block

从嵌套的块列表组装 nd 数组。

split

将数组拆分成多个大小相等的子数组列表。

备注

unstack 充当 stack 的反向操作，即 stack(unstack(x, axis=axis), axis=axis) == x。

此函数等效于 tuple(np.moveaxis(x, axis, 0))，因为迭代数组会沿着第一轴迭代。

示例

>>> arr = np.arange(24).reshape((2, 3, 4))
>>> np.unstack(arr)
(array([[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]]),
 array([[12, 13, 14, 15],
        [16, 17, 18, 19],
        [20, 21, 22, 23]]))
>>> np.unstack(arr, axis=1)
(array([[ 0,  1,  2,  3],
        [12, 13, 14, 15]]),
 array([[ 4,  5,  6,  7],
        [16, 17, 18, 19]]),
 array([[ 8,  9, 10, 11],
        [20, 21, 22, 23]]))
>>> arr2 = np.stack(np.unstack(arr, axis=1), axis=1)
>>> arr2.shape
(2, 3, 4)
>>> np.all(arr == arr2)
np.True_
本页内容