История изменений

Написать желаемое ТС можно разными способами: чем тривиальнее задача, тем больше способов ее решить, как в его случае. Что-то очень сложное наоборот решить можно зачастую лишь единичным образом.
Но поскольку код будет читать и развивать дальше именно ТС, а не мы, логичнее подправить его собственный код, вместо предложения наших вариантов, пусть и даже на голову лучших, допустим — свой-то код он и после чужой доработки поймет.

Можно ли в питоне конкатенировать имена переменных или методов? (комментарий) — дельные мысли, однако в случае ТС либо он пишет кот, либо учится писать на Питоне, — выбирайте любые один 🙃.

Написать желаемое ТС можно разными способами: чем тривиальнее задача, тем больше способов ее решить, как в его случае. Что-то очень сложное наоборот решить можно зачастую лишь единичным образом.
Но поскольку код будет читать и развивать дальше именно ТС, а не мы, логичнее подправить его собственный код, вместо предложения наших вариантов, пусть и даже на голову лучших, допустим — свой-то код он и после чужой доработки поймет.