Python(Numpy)————数组堆叠与分割

Myth丶恋晨 2022-12-31 05:23 139阅读 0赞

## 堆叠与分割 ##

--------------------

### 1 水平方向 ###

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ppYW5neWlfMTYxMjEwMV8wMw_size_16_color_FFFFFF_t_70]

import numpy as np
    
    
    # 1.水平堆叠(需要行相同)
    arr1 = np.zeros(shape=(2, 2), dtype=int)
    print(arr1)
    arr2 = np.ones(shape=(2, 3), dtype=int)
    print(arr2)
    ret = np.hstack((arr1, arr2))       # 注意括号(arr1, arr2)
    print(ret)
    
    # 1 水平分割(分割数需要是列的商数)
    arr = np.arange(16).reshape(4, 4)
    ret = np.hsplit(arr, 2)
    print(ret)

--------------------

### 2 竖直方向 ###

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ppYW5neWlfMTYxMjEwMV8wMw_size_16_color_FFFFFF_t_70 1]

# 2.垂直堆叠(需要列相同)
    arr1 = np.ones(shape=(2, 3))
    arr2 = np.zeros(shape=(3, 3))
    ret = np.vstack((arr1, arr2))
    print(ret)
    
    # 2. 垂直分割
    arr = np.arange(8).reshape(4, 2)
    ret = np.vsplit(arr, 2)
    print(ret)

--------------------

### 3 深度方向 ###

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ppYW5neWlfMTYxMjEwMV8wMw_size_16_color_FFFFFF_t_70 2]

red = np.arange(0, 9)
    green = np.arange(9, 18)
    blue = np.arange(18, 27)
    ret = np.dstack((red, green, blue))
    print(ret)
    
    # 3 深度分割
    arr = np.arange(16).reshape(2, 2, 4)
    ret = np.dsplit(arr, 2)
    print(ret)

--------------------

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ppYW5neWlfMTYxMjEwMV8wMw_size_16_color_FFFFFF_t_70]: /images/20221120/e024410f549b4d829e9f635184c6441e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ppYW5neWlfMTYxMjEwMV8wMw_size_16_color_FFFFFF_t_70 1]: /images/20221120/05aa737816df48bb936f61f8d5b04a07.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ppYW5neWlfMTYxMjEwMV8wMw_size_16_color_FFFFFF_t_70 2]: /images/20221120/e70c877e292544a0836805af8438a45d.png