numpy.split() 函数
此函数将数组沿指定轴划分为子数组。 该函数采用三个参数。
numpy.split(ary, indices_or_sections, axis)
这里,
| 序号 | 参数 & 描述 |
|---|---|
| 1 | ary 要拆分的输入数组 |
| 2 | indices_or_sections 可以是一个整数，表示要从输入数组中创建的大小相等的子数组的数量。 如果此参数是一维数组，则条目指示要创建新子数组的点。 |
| 3 | axis 默认为 0 |
示例
import numpy as np a = np.arange(9) print 'First array:' print a print '\n' print 'Split the array in 3 equal-sized subarrays:' b = np.split(a,3) print b print '\n' print 'Split the array at positions indicated in 1-D array:' b = np.split(a,[4,7]) print b
它的输出结果如下 −
First array: [0 1 2 3 4 5 6 7 8] Split the array in 3 equal-sized subarrays: [array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8])] Split the array at positions indicated in 1-D array: [array([0, 1, 2, 3]), array([4, 5, 6]), array([7, 8])]
