Update 01.py

if文をスマートにしました！
Merge pull request #1 from pika-shi/patch-1

しゃあなしやで