极客教程NumPy 数组操作 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])]