Можно ли упростить функцию подсчета, убрав лишние переменные и сразу возвращать результат в Python

кодоптимизацияпроизводительностьупрощениефункции