Python numpy.vsplit函数方法的使用

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)