Numpy-split

numpy.split

この関数は、指定された軸に沿って配列をサブ配列に分割します。 この関数は3つのパラメーターを取ります。

numpy.split(ary, indices_or_sections, axis)

どこで、

Sr.No.	Parameter & Description
1	ary 分割する入力配列
2	indices_or_sections 入力配列から作成される同じサイズのサブ配列の数を示す整数を指定できます。 このパラメーターが1次元配列の場合、エントリは新しいサブ配列が作成されるポイントを示します。
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])]