python数组分割为多份,python – 以多个值分割numpy数组？

1d案例可以像这样完成

>>> A = np.array([0.1, 3.5, 6.5, 7.9, 11.4, 12.0, 22.3, 24.5, 26.7, 29.9])

>>> split_at = A.searchsorted([10, 20])

>>> B = numpy.split(A, split_at)

如果我正确理解你的问题,这也适用于2d,例如：

>>> A = array([[ 0.1, 0. ],

[ 3.5, 1. ],

[ 6.5, 2. ],

[ 7.9, 3. ],

[ 11.4, 4. ],

[ 12. , 5. ],

[ 22.3, 6. ],

[ 24.5, 7. ],

[ 26.7, 8. ],

[ 29.9, 9. ]])

>>> split_at = A[:, 0].searchsorted([10, 20])

>>> B = numpy.split(A, split_at)

>>> B

[array([[ 0.1, 0. ],

[ 3.5, 1. ],

[ 6.5, 2. ],

[ 7.9, 3. ]]),

array([[ 11.4, 4. ],

[ 12. , 5. ]]),

array([[ 22.3, 6. ],

[ 24.5, 7. ],

[ 26.7, 8. ],

[ 29.9, 9. ]])]