python题列表啊哦嘿喂

编写程序，先输入一个含有整数元素的列表，在输入一个整数数据，判断该数据是否为列表中的元素。若是删掉第一个数据，否则，将数据插入到列表，使成为首元素

a = [int(n) for n in input().split()]
b = int(input())
if b in a:
    a.pop(0)
else:
    a.insert(0, b)
print(a)