Есть массив, после сортировки надо сохранить первоначальные индексы

data1=np.random.randint(1,22,(10,4))
#не сортированный= data1  
[[ 2 19  7 18]  
 [18  3 12  1]  
 [ 7 10 19 20]  
 [ 6 16  2 19] 
 [ 8  6 18 17] 
 [ 2 13 11 16]  
 [ 8  1  6 10] 
 [18  9  2  7] 
 [ 8 15 19  2]
 [ 3  4  9  7]] 
 
num1=len(data1) # 10 строк
data2=np.sort(data1[0:num1,0:4]) #сортируем
 
#сортированный = data2
[[ 2  7 18 19]
 [ 1  3 12 18]
 [ 7 10 19 20]
 [ 2  6 16 19]
 [ 6  8 17 18]
 [ 2 11 13 16]
 [ 1  6  8 10]
 [ 2  7  9 18]
 [ 2  8 15 19]
 [ 3  4  7  9]]

for i in range(num1):
    data0=sorted(enumerate(data1[i]), key=lambda j: j[1]) # собираем индексы
    data3.extend(data0) #получаем массив "data 3" состоящий из индекса и числа
 
#[(0, 2), (2, 7), (3, 18), (1, 19), (3, 1), (1, 3), (2, 12), (0, 18), (0, 7), (1, 10), (2, 19), (3, 20), (2, 2), (0, 6), 
#(1, 16), (3, 19), (1, 6), (0, 8), (3, 17), (2, 18), (0, 2), (2, 11), (1, 13), (3, 16), (1, 1), (2, 6), (0, 8), (3, 10), 
#(2, 2), (3, 7), (1, 9), (0, 18), (3, 2), (0, 8), (1, 15), (2, 19), (0, 3), (1, 4), (3, 7), (2, 9)]
#до этого момента как бы все гладко,а вот дальше )))
 
data4= (data3[:1])   # получаем [(0, 2)] первый номер из массива индексов
data4=(' '.join(map(str, data4)))# превращаю в строку символов
s=(data4[4:5])  # получаю число 2 из первой ячейки
cc=(data2[0:num1,0:1]) #  срез первого столбца сортированного массива
 
#далее поиски решения привели к пшику))), провожу сравнение числа в несортированном  
#и сортированном массиве что бы "забрать" индекс
for i in range(num1):
   f=cc[:1]
   f=' '.join(map(str, f))
   if s==f[1:2]:
      print('Yes')
 #далее ??????????????????????? пока не знаю как далее

import numpy as np 
 
a = np.array([[2,1,3,4],[7,6,5,8],[10,9,12,11]])
b = a.argsort()
rows, cols = a.shape
a = np.sort(a,axis=1)
 
c = np.vstack((a.ravel(),b.ravel())).T.reshape(rows, cols, 2)
res = np.hstack(c).reshape(cols,rows,2)
print(*res, sep='\n')

import numpy as np 
 
a = np.array([[2,1,3,4],[7,6,5,8],[10,9,12,11]])
 
rows, cols = a.shape
res = np.empty((cols,rows,2), dtype=np.int8)
res[:,:,1] = a.argsort().T
res[:,:,0] = np.sort(a,axis=1).T
print(*res, sep='\n')

arr=[6,0,-2,3,7,1,12,0,-7]
tmp=list(zip(arr,range(len(arr))))
for a in sorted(tmp):
    print(*a)