numpy.vsplit 函数用于沿着垂直方向(即沿着行的方向)将一个数组分割成多个子数组。它将原始数组根据指定的索引数组或数量进行分割,返回一个包含多个子数组的列表。本文主要介绍一下NumPy中vsplit方法的使用。

numpy.vsplit

numpy.vsplit(ary, indices_or_sections)     [source]

垂直(行)将数组拆分为多个子数组。

请参考split文档。vsplit等价于axis=0(默认)进行split,无论数组的维度如何,数组总是沿着第一个轴进行分割。

例子

1)按数量分割

如果指定一个整数 Nnumpy.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)

推荐文档

相关文档

大家感兴趣的内容

随机列表