こんな感じのダサいけど読みやすいコードじゃダメなん? Pythonユーザーはこんな感じで書く人が多いんじゃないかなと思うんだけど。
https://www.ideone.com/h3xXoq