В питоне есть красивые решения для типовых небольших задач.)
что бы их знать, нужно изучать различные кукбуки.
также библиотеки которые входят в стандартный набор.
т.е. изучение лямбд, [], {}, map ... На примерах и из документации.
библиотеки в первую очередь сollections, Functional Programming Module.
Через это придет опыт написания красивого и выразительного кода.
Касаемо, как писать вообще код (не только решение мелких типовых задач), то здесь на первое место выходит умение выбрать хорошее название (переменной, функции, класса и т.д.) Про это есть много книг и видео. Это очень важно.