我要一段烟火代码Python

好的，以下是一个简单的Python代码，可以绘制出一个基本的烟火效果：

import turtle
import random

# 设置画布大小
screen = turtle.Screen()
screen.setup(600, 600)

# 定义烟花颜色列表
colors = ['red', 'blue', 'green', 'purple', 'yellow']

# 定义烟花形状列表
shapes = ['circle', 'square', 'turtle', 'triangle', 'classic']

# 定义烟花数量
num_fireworks = 10

# 循环生成烟花
for i in range(num_fireworks):
    # 随机选择颜色和形状
    color = random.choice(colors)
    shape = random.choice(shapes)
    
    # 创建烟花画笔
    pen = turtle.Turtle()
    pen.hideturtle()
    pen.speed(0)
    pen.color(color)
    pen.shape(shape)
    
    # 随机生成烟花起点和终点
    x = random.randint(-200, 200)
    y = random.randint(-200, 200)
    end_x = random.randint(-200, 200)
    end_y = random.randint(-200, 200)
    
    # 移动画笔到起点
    pen.penup()
    pen.goto(x, y)
    pen.pendown()
    
    # 绘制烟花
    pen.goto(end_x, end_y)
    
    # 模拟烟花爆炸效果
    for j in range(10):
        pen.color(random.choice(colors))
        pen.stamp()
        pen.right(36)
        
    # 清空画笔
    pen.clear()

这个代码使用了turtle模块来绘制烟花效果。每个烟花具有随机的颜色和形状，并随机生成起点和终点。绘制烟花时，画笔从起点移动到终点，然后模拟爆炸效果。最后清空画笔。你可以根据需要修改颜色、形状、数量等参数来定制你需要的烟花效果。