np的split

import  numpy as np

'''
split(ary, indices_or_sections, axis=0):
(1)如果' indices_or_sections '是整数N，则数组将被分割
沿“轴”排列成N个相等的数组。如果这样的分裂是不可能的，
出现错误。
(2)如果' indices_or_sections '是一个由已排序整数组成的一维数组，则输入
指示数组沿“轴”的分割位置。例如,
' '[2,3] ' '对于' ' axis=0 ' '，将导致
          - ary[:2]
          - ary[2:3]
          - ary[3:]
Examples
    --------
    >>> x = np.arange(9.0)
    >>> np.split(x, 3)
    [array([ 0.,  1.,  2.]), array([ 3.,  4.,  5.]), array([ 6.,  7.,  8.])]

    >>> x = np.arange(8.0)
    >>> np.split(x, [3, 5, 6, 10])
    [array([ 0.,  1.,  2.]),
     array([ 3.,  4.]),
     array([ 5.]),
     array([ 6.,  7.]),
     array([], dtype=float64)]
'''
x = np.arange(8.0)
print(np.split(x, [7,8]))
print(np.split(x, (7,)))
# 是第（2）种情况
# [array([0., 1., 2., 3., 4., 5., 6.]), array([7.]), array([], dtype=float64)]
# [array([0., 1., 2., 3., 4., 5., 6.]), array([7.])]