numpy.vsplit
numpy.vsplit(ary, indices_or_sections) [source]
垂直(行)将数组拆分为多个子数组。
请参考split文档。vsplit
等价于axis=0
(默认)进行split,无论数组的维度如何,数组总是沿着第一个轴进行分割。
例子
1)按数量分割
如果指定一个整数 N
,numpy.vsplit
会将输入数组均匀地分割成 N
个部分。
import numpy as np arr = np.arange(16).reshape(4, 4) print("原始数组:") print(arr) # 将数组均匀分割成2个子数组 result = np.vsplit(arr, 2) print("\n分割后的子数组:") for subarr in result: print(subarr)
2)按具体的分割位置分割
可以通过一个索引数组指定具体的分割位置。
import numpy as np arr = np.arange(16).reshape(4, 4) print("原始数组:") print(arr) # 按指定的索引进行分割 result = np.vsplit(arr, [2]) print("\n分割后的子数组:") for subarr in result: print(subarr)