def first_true2():
    di = {}
for i in range(len(a)):
    idx = np.argmax(a[i])
if idx > 0:
    di[i] = idx
else:
    di[i] = None
return di
Поиск первого ненулевого значения по оси отсортированного двумерного массива numpy