【Numpy】拆分数组

前言

本篇总结、介绍Numpy数组（ndarray）的基本操作之一——拆分数组 [1]。

1. split

numpy.split(ary, indices_or_sections, axis=0)：将一个数组拆分为多个子数组，每个子数组都是原始数组的视图

• ary：类数组。待拆分数组
• indices_or_sections：整数或一维类数组
  • 整数：例如n，表示数组ary将沿指定的轴axis拆分为n个相等的数组。如果该拆分不可行，那么会引起错误
  • 一维类数组（如列表、元组）：指示沿着指定的轴axis拆分的位置。如numpy.split(ary,[2,3], axis=0)将返回[ary[:2], ary[2:3], ary[3:]]
• axis：整数型，可选参数

>>> arr = np.arange(8)
>>> arr
array([0, 1, 2, 3, 4, 5, 6, 7])

>>> np.split(arr,2)
[array([0, 1, 2, 3]), array([4, 5, 6,