自用--np.stack(), np.vstack(), np.hstack()

最新推荐文章于 2024-04-27 20:58:54 发布

P0ny

最新推荐文章于 2024-04-27 20:58:54 发布

阅读量466

点赞数 1

分类专栏： numpy python

本文链接：https://blog.csdn.net/qq_36735489/article/details/93176664

版权

python 同时被 2 个专栏收录

4 篇文章 0 订阅

订阅专栏

numpy

1 篇文章 0 订阅

订阅专栏

np.stack(list or tuple , axis=0, out=None)
axis写的是哪个维度,就在哪个维度插入叠加的数组的总数

    >>> arrays = [np.random.randn(3, 4) for _ in range(10)] # 产生一个list,包含10个shape为(3,4)的数组
    >>> np.stack(arrays, axis=0).shape  # 共10个, 10插入第0维
    (10, 3, 4)

    >>> np.stack(arrays, axis=1).shape # 共10个, 10插入第1维
    (3, 10, 4)

    >>> np.stack(arrays, axis=2).shape # 共10个, 10插入第2维
    (3, 4, 10)

np.vstack(list or tuple)
垂直堆叠数组

>>> a = np.array([1, 2, 3])  # 
>>> b = np.array([2, 3, 4])
>>> np.vstack((a,b)) # 两个[1,3]垂直堆叠 --> [2,3]
array([[1, 2, 3],
       [2, 3, 4]])

>>> a = np.array([[1], [2], [3]]) # 两个[3,1]垂直堆叠 --> [6,1]
>>> b = np.array([[2], [3], [4]])
>>> np.vstack((a,b))
array([[1],
       [2],
       [3],
       [2],
       [3],
       [4]])

np.vstack(list or tuple)
水平堆叠数组

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