写一段智能画图的python程序，能画爱心，正方形，圆形和五角星，能询问我要画什么

我理解了你的要求，请看下面的代码：

import turtle

print("你想画什么图形？请输入数字：")
print("1. 爱心")
print("2. 正方形")
print("3. 圆形")
print("4. 五角星")

choice = input("请输入数字：")

if choice == "1":
    # 画爱心
    turtle.color('red')
    turtle.begin_fill()
    turtle.left(45)
    turtle.forward(100)
    turtle.circle(50, 180)
    turtle.right(90)
    turtle.circle(50, 180)
    turtle.forward(100)
    turtle.end_fill()
elif choice == "2":
    # 画正方形
    turtle.color('blue')
    for i in range(4):
        turtle.forward(100)
        turtle.right(90)
elif choice == "3":
    # 画圆形
    turtle.color('green')
    turtle.begin_fill()
    turtle.circle(50)
    turtle.end_fill()
elif choice == "4":
    # 画五角星
    turtle.color('yellow')
    turtle.begin_fill()
    for i in range(5):
        turtle.forward(100)
        turtle.right(144)
    turtle.end_fill()
else:
    print("输入错误，请输入数字 1-4。")

turtle.exitonclick()

运行程序后，会询问你要画什么图形，输入对应的数字即可绘制相应的图形。注意，这里用到了 `turtle` 库，需要保证已经安装。