用python编摇色子

# author : momo
#提示:押大还押小  ： 大  或者 小
# 开始摇色子，
#
# 【1~6】取值【1， 2， 3】 小
#
#  取值【4， 5， 6】大
#
# 若押中，则打印“”恭喜你中了五百万“
#
# 若没押中，则打印”谢谢惠顾！“
import random
a=input("押大还是小：")
b=random.choice(range(1,7))
if (b<4) and (a=="小") or (b>=4) and (a=="大"):
    print("数字是：",b)
    print("恭喜您中了500万")
else:
    print("数字是：", b)
    print("谢谢惠顾")