能写一个难一点的python烟花代码吗

当然可以！这是一个使用 Python 语言编写的烟花效果的示例代码：

import random
import math
import pygame

# 定义窗口的宽度和高度
WIDTH, HEIGHT = 800, 600

# 初始化 Pygame
pygame.init()

# 创建窗口
screen = pygame.display.set_mode((WIDTH, HEIGHT))

# 定义颜色
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
RED = (255, 0, 0)
GREEN = (0, 255, 0)
BLUE = (0, 0, 255)
YELLOW = (255, 255, 0)
CYAN = (0, 255, 255)
MAGENTA = (255, 0, 255)

# 定义粒子类
class Particle:
    def __init__(self, x, y, size, color, angle, speed):
        self.x = x
        self.y = y
        self.size = size
        self.color = color
        self.angle = angle
        self.speed = speed
        self.gravity = 0.1
        self.fade = random.uniform(0.005, 0.01)

    def update(self):
        # 根据角度和速度更新粒子的位置
        self.x += math.cos(self.angle) * self.speed
        self.y += math.sin(self.angle) * self.speed
        self.speed *= 1 - self.fade
        self.y += self.gravity

    def draw(self, surface):
        # 绘制粒子
        pygame.draw.circle(surface, self.color, (int(self.x), int(self.y)), self.size)

# 定义烟花类
class Firework:
    def __init__(self, x, y, color):
        self.x = x
        self.y = y
        self.color = color
        self.particles = []
        self.exploded = False
        self.lifetime = random.uniform(0.6, 1.2)

    def update(self):
        if not self.exploded:
            # 如果烟花没有爆炸，则向上攀升
            self.y -= 5
            if self.lifetime <= 0:
                # 如果烟花的寿命已经结束，则爆炸
                self.explode()
        else:
            # 如果烟花已经爆炸