📅  最后修改于: 2023-12-03 15:18:03.184000             🧑  作者: Mango
在使用 numpy 处理数组时,经常需要将数组拆分为大小相等的块进行分析或操作。numpy 提供了 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 数组按照指定的块数进行拆分,以及如何通过设置轴参数来指定拆分方向。拆分后会返回拆分后的子数组列表,方便进行下一步的操作和分析。