for i in range(len(a)):
    idx = np.where(a[i] > 0)
try:
    di[i] = idx[0][0]
except IndexError:
    di[i] = None
return di
Поиск первого ненулевого значения по оси отсортированного двумерного массива numpy