沿轴分离数组 - Python

在Python中，我们可以使用numpy库中的split函数来沿着指定的轴将数组进行分离。该函数将一个数组沿着某个轴分离成多个子数组。

函数原型

numpy.split(arr, indices_or_sections, axis=0)

• arr：要分离的数组。
• indices_or_sections：可以是一个整数，表示要分离成的子数组数量；也可以是一组用于指定分离点的整数（如 [2, 3] 代表在第2个和第3个元素之间分离）。
• axis：要分离的轴，默认为0，即第一个轴。

使用示例

import numpy as np

arr = np.arange(10).reshape(5, 2)
print("原数组：\n", arr)

# 沿着第一个轴将数组分离成2个子数组
result = np.split(arr, 2, axis=0)
print("分离后的子数组1：\n", result[0])
print("分离后的子数组2：\n", result[1])

输出结果：

原数组：
 [[0 1]
  [2 3]
  [4 5]
  [6 7]
  [8 9]]
分离后的子数组1：
 [[0 1]
  [2 3]
  [4 5]]
分离后的子数组2：
 [[6 7]
  [8 9]]

在上述示例中，我们使用split函数将长度为5的数组沿着第一个轴分离成了两个长度为3和2的子数组。

注意事项

• 分离前数组的总长度必须刚好能被分离成的子数组数量整除，否则会抛出ValueError异常。
• 分离后的子数组与原数组共享数据存储空间，因此改变其中任何一个也会影响其他的数组。
• 可以将split函数与hstack、vstack等函数结合使用，实现更加灵活的数组操作。