Cara, parabéns!

Não é algo bobo, leva tempo para termos insights sobre essas coisas e nesse caso que você mostrou ao usar o loop.first é simplesmente fenomenal!

Ano passado escrevi um post em que mostro alguns "truques" para encurtar algumas coisas em Python: https://niumxp.github.io/python/one-line; da uma passadinha lá, vai que mais alguma coisa te inspire 👀