Python numpy.hsplit函数方法的使用

numpy.hsplit

numpy.hsplit(ary, indices_or_sections)     [source]

水平（按列）将一个数组拆分为多个子数组。

请参考split文档。hsplit等同于axis = 1的split，无论数组的维度如何，数组总是沿着第二个轴分割。

例子

1）均匀拆分

import numpy as np

arr = np.arange(12).reshape(3, 4)  # 生成一个 3x4 的数组
print("原数组：\n", arr)

# 将数组水平拆分为 2 份
result = np.hsplit(arr, 2)

print("拆分结果：")
for i, sub_arr in enumerate(result):
    print(f"部分 {i+1}:\n", sub_arr)

2）按指定列索引拆分

import numpy as np

arr = np.arange(12).reshape(3, 4)
result = np.hsplit(arr, [1, 3])  # 在索引 1 和 3 处拆分

print("拆分结果：")
for i, sub_arr in enumerate(result):
    print(f"部分 {i+1}:\n", sub_arr)