ご回答ありがとうございます。こちらの書き方もスマートで良いですね！
まだif~switch文や配列までは手が及んでいなかったので、リンク先のHPは大変参考になりそうです。少しずつ読んでいきたいと思います。