if i == 0


else i == 1


else


이런식으로 했으면 더 간단해 지지 않았을까요?