给我写一段植物大战僵尸代码

当然可以！以下是一个简单的植物大战僵尸游戏的示例代码：

class Plant:
    def __init__(self, name, cost, attack):
        self.name = name
        self.cost = cost
        self.attack = attack

class Zombie:
    def __init__(self, name, health, attack):
        self.name = name
        self.health = health
        self.attack = attack

def main():
    # 创建植物
    sunflower = Plant("Sunflower", 50, 0)
    peashooter = Plant("Peashooter", 100, 25)

    # 创建僵尸
    normal_zombie = Zombie("Normal Zombie", 100, 10)
    conehead_zombie = Zombie("Conehead Zombie", 200, 20)

    # 游戏主循环
    while True:
        print("请选择要种植的植物：")
        print("1. Sunflower (阳光花) - 50阳光")
        print("2. Peashooter (豌豆射手) - 100阳光")
        print("3. 结束游戏")

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

        if choice == "1":
            print("您选择了种植 Sunflower！")
            # 在这里编写种植 Sunflower 的逻辑

        elif choice == "2":
            print("您选择了种植 Peashooter！")
            # 在这里编写种植 Peashooter 的逻辑

        elif choice == "3":
            print("游戏结束！")
            break

        else:
            print("无效选项，请重新选择！")

if __name__ == "__main__":
    main()

请注意，这只是一个简单的示例代码，您可以根据自己的需求和游戏规则进行进一步的扩展和完善。希望对您有所帮助，祝您编写出一个有趣的植物大战僵尸游戏！