python代码也不会写 numpy

Question

假如我有个区间
a=np.array([1,2,3,4,5,6,7,8,9,10])
b=np.array([3,7])
我怎么才可以把向量a分成三类
一类为a1=[1,2,3] 一类为a2=[3,4,5,6,7]，最后一类为a3=[7,8,9,10]呢
就是怎么用b来分割a呀？
求解答

Answer 1

import numpy as np
import more_itertools as mi


a=np.array([1,2,3,4,5,6,7,8,9,10])
b=np.array([3,7])

res = list(mi.split_when(a, lambda x, y: x in b))
result = [res[i] if i == 0 else [b[i-1]] + res[i] for i in range(len(res ))]
print(result)