把input改为raw_input即可
这是因为input()返回的数据类型是str，str不能直接和整数比较，必须先把str转换成整数。

以下来自https://blog.csdn.net/duke10/article/details/78415375
区别：
input：会根据用户的输入来做类型的转换
raw_input：则会把用户的输入都作为一个字符串来处理
为了引起不必要的麻烦，推荐使用raw_input
以下来自https://www.cnblogs.com/noxy/p/6362614.html
问题原因：
　　对于 input() ，它希望能够读取一个合法的 python 表达式，即你输入字符串的时候必须使用引号将它括起来，否则它会引发一个 SyntaxError 。
扩展：
　　关于python中的raw_input()和input()，两者都是python 的内建函数，通过读取控制台的输入与用户实现交互，主要区别有如下两点：
1、这两个函数均能接收 字符串 ，但 raw_input() 直接读取控制台的输入（任何类型的输入它都可以接收）。而对于 input() ，它希望能够读取一个合法的 python 表达式，即你输入字符串的时候必须使用引号将它括起来，否则它会引发一个 SyntaxError 。
2、raw_input() 将所有输入作为字符串看待，返回字符串类型。而 input() 在对待纯数字输入时具有自己的特性，它返回所输入的数字的类型（ int, float ）；同时在例子 1 知道，input() 可接受合法的 python 表达式，举例：input( 1 + 3 ) 会返回 int 型的
从help(input)的帮助结果可知，input()本质上还是通过raw_input()实现的，只是调用完 raw_input() 之后再调用 eval() 函数，所以，你甚至可以将表达式作为 input() 的参数，并且它会计算表达式的值并返回它。
不过在 Built-in Functions 里有一句话是这样写的：Consider using the raw_input() function for general input from users.除非对 input() 有特别需要，否则一般情况下我们都是推荐使用 raw_input() 来与用户交互。

Help on built-in function input in module __builtin__:
 
input(...)
    input([prompt]) -> value
     
    Equivalent to eval(raw_input(prompt)).
(END)