分支语句中的相关编程问题

s = input('请输入图形编号')
if s == '1':
    print('正方形')
elif s == '2':
    print('三角形')
elif s == '3':
    print('圆形')
else:
    print('请重新运行程序！')