Оптимизированный способ сделать это в Python с enumerate