def get_jarvis(x,y): jarvis = [] for i in range(len(x)): if x[i] == y[0]: jarvis.append([y[0],y[1]]) elif x[i] == y[1]: jarvis.append([y[1],y[2]]) elif x[i] == y[2]: jarvis.append([y[2],y[3]]) else: pass return np.array(jarvis).astype('float32')