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. ]])]