Numpy 将数组拆分为大小相等的块 - Python

在使用 numpy 处理数组时，经常需要将数组拆分为大小相等的块进行分析或操作。numpy 提供了 array_split 函数，它可以将数组按照指定的块数进行拆分，并返回拆分后的子数组列表。本文将介绍如何使用 array_split 函数拆分数组，并提供示例代码。

array_split 函数的语法

array_split 函数的语法如下：

numpy.array_split(ary, indices_or_sections, axis=0)

其中，ary 表示要拆分的数组；indices_or_sections 可以是一个整数，表示要拆分的块数；也可以是一个整数列表，表示要在哪些位置拆分数组；axis 表示拆分的方向，即沿着哪个轴进行拆分。默认为 0，表示沿着第一个轴（行）进行拆分。

示例代码

下面是一个示例，演示如何使用 array_split 函数将数组按照指定的块数进行拆分：

import numpy as np

# 定义一个二维数组
arr = np.arange(15).reshape((3, 5))

# 将数组按照块数拆分
sub_arr_list = np.array_split(arr, 2)

# 输出拆分后的子数组列表
for sub_arr in sub_arr_list:
    print(sub_arr)

上述代码将定义一个包含 3 行、5 列元素的二维数组。然后，使用 array_split 函数将数组按照 2 个块进行拆分，并将拆分后的子数组列表打印输出。

运行结果如下：

[[0 1 2]
 [3 4 5]]
[[ 6  7  8]
 [ 9 10 11]
 [12 13 14]]

总结

本文介绍了如何使用 array_split 函数将 numpy 数组按照指定的块数进行拆分，以及如何通过设置轴参数来指定拆分方向。拆分后会返回拆分后的子数组列表，方便进行下一步的操作和分析。