求大佬用python帮忙看一下这道题

求大佬用python帮忙看一下这道题

这个简单。。。我写个大概，没有上机测试，有问题你自己改改吧

vote_input=input("输入投票结果")
vote=vote_input.split(",")

solve1=set(vote)
print(solve1) #候选人序号


solve2=[]
for i in solve1:
    if i in ["1","2","3","4"]:
        solve2.append(i)
print(solve2) #1-4获得选票

solve3=[]
for i in solve1:
    if i not in ["5","6","7","8"]:
        solve3.append(i)
print(solve3) #5-8没有获得选票


solve4=input("输入候选人")
if solve4 in solve1:
    print("True")
else:
    print("False")