numpy array split

import numpy as np


# 创建数组
arr = np.arange(16).reshape(4,4)
# print(arr)

# 横向拆分
# arr1 = np.hsplit(arr, 4)
# print(arr1)
arr2 = np.split(arr, 4, axis=1)
# print(arr2)

# 纵向拆分
arr3 = np.vsplit(arr , 4)
print(arr3)
arr4 = np.split(arr, 4,axis=0)
# print(arr4)

Guess you like

Origin blog.csdn.net/YPL_ZML/article/details/93239386