def first_true3():
    di = {}
for i, j in zip(*np.where(a > 0)):
    if i in di:
        continue
else:
    di[i] = j
return di
Поиск первого ненулевого значения по оси отсортированного двумерного массива numpy