如何在Py3.5中搜索一维列表中的元素

Question

我正在尝试编写一个基本程序，该程序可以告诉用户他们输入的元素是否存在于他们输入的列表中。

我的代码：

a=(int(x) for x in input().split())
b=input('Enter the element you want to see if is present or not')
if b in a:
   print('Yes it is present')
else:
   print('Try again')

输出：再试一次（在所有情况下）

请帮我纠正这段代码，并告诉我为什么if b in a不起作用，但b in a在Python shell中有效。

感谢：）