Функциональность, которую я ищу, выглядит примерно так:
data = np.array([[1, 2, 3, 4],
[2, 3, 1],
[5, 5, 5, 5],
[1, 1]])
result = fix(data)
print result
[[ 1. 2. 3. 4.]
[ 2. 3. 1. 0.]
[ 5. 5. 5. 5.]
[ 1. 1. 0. 0.]]
Эти массивы данных, с которыми я работаю, действительно большие, поэтому я бы очень оценил наиболее эффективное решение.
Изменить: данные считываются с диска в виде списка списков python.