手把手教你用python多线程绘制跨年烟花

 

手把手教你用python多线程绘制跨年烟花

在本项目中，我们使用了 `threading` 库创建了一个新线程，在该线程中绘制跨年烟花。我们使用 `matplotlib` 库的 `scatter` 函数绘制跨年烟花，并使用 `plt.pause` 函数暂停一段时间，再使用 `clear` 函数清除图像。

 

 代码：

import random
import threading
import time
import imageio

import matplotlib.pyplot as plt
import numpy as np


# 绘制跨年烟花的函数
def draw_fireworks(ax):
    # 绘制烟花的颜色
    colors = ['r', 'y', 'b', 'g', 'm', 'c']
    # 随机选择烟花的颜色
    color = random.choice(colors)
    
    # 随机生成烟花的爆炸位置
    x = random.uniform(-1, 1)
    y = random.uniform(-1, 1)
    
    # 绘制烟花的爆炸效果
    ax.scatter(x, y, color=color, s=100)
    
    # 随机生成烟花的碎片数量
    num_fragments = random.randint(10, 20)
    
    # 随机生成烟花的碎片位置
    for _ in range(num_fragments):
        x = random.uniform(-1, 1)
        y = random.uniform(-1, 1)
        ax.scatter(x, y, color=color, s=10)

# 创建一个图像列表，用于保存绘制的每一帧
images = []

# 绘制跨年烟花的线程
def fireworks_thread():
    # 创建一个新的图像
    fig, ax = plt.subplots()
    
    # 设置图像的背景色为黑色
    # 设置图像的背景色为黑色
    ax.set_facecolor('k')

    # 循环绘制跨年烟花
    while True:
        # 绘制跨年烟花
        draw_fireworks(ax)
        
        # 暂停一段时间
        plt.pause(0.1)
        
        # 清除图像
        ax.clear()
thread = threading.Thread(target=fireworks_thread)
thread.start()
time.sleep(5)
thread.stop()
time.sleep(5)

imageio.mimsave('fireworks.gif', images)
       

 效果发现不理想，烟花太稀疏。

 我们可以改进一下：

如果希望让烟花放的更多，可以在绘制跨年烟花的函数 draw_fireworks 中增加循环次数。例如，可以在线程中使用以下代码段改变烟花的数量：

# 绘制跨年烟花的线程
def fireworks_thread():
    # 创建一个新的图像
    fig, ax = plt.subplots()
    
    # 设置图像的背景色为黑色
    ax.set_facecolor('k')
    
    # 循环绘制跨年烟花
    while True:
        # 绘制若干个跨年烟花
        for _ in range(5):
            # 绘制跨年烟花
            draw_fireworks(ax)
        
        # 暂停一段时间
        plt.pause(0.1)
        
        # 清除图像
        ax.clear()

在上述代码中，我们在每次循环中绘制了 5 个跨年烟花。您可以根据自己的需要调整这个数字，使烟花的数量更多。

效果图：

 

完整代码：

import random
import threading
import time
import imageio

import matplotlib.pyplot as plt
import numpy as np


# 绘制跨年烟花的函数
def draw_fireworks(ax):
    # 绘制烟花的颜色
    colors = ['r', 'y', 'b', 'g', 'm', 'c']
    # 随机选择烟花的颜色
    color = random.choice(colors)
    
    # 随机生成烟花的爆炸位置
    x = random.uniform(-1, 1)
    y = random.uniform(-1, 1)
    
    # 绘制烟花的爆炸效果
    ax.scatter(x, y, color=color, s=100)
    
    # 随机生成烟花的碎片数量
    num_fragments = random.randint(10, 20)
    
    # 随机生成烟花的碎片位置
    for _ in range(num_fragments):
        x = random.uniform(-1, 1)
        y = random.uniform(-1, 1)
        ax.scatter(x, y, color=color, s=10)

# 创建一个图像列表，用于保存绘制的每一帧
images = []

# 绘制跨年烟花的线程
# 绘制跨年烟花的线程
def fireworks_thread():
    # 创建一个新的图像
    fig, ax = plt.subplots()
    
    # 设置图像的背景色为黑色
    ax.set_facecolor('k')
    
    # 循环绘制跨年烟花
    while True:
        # 绘制若干个跨年烟花
        for _ in range(5):
            # 绘制跨年烟花
            draw_fireworks(ax)
        
        # 暂停一段时间
        plt.pause(0.1)
        
        # 清除图像
        ax.clear()

thread = threading.Thread(target=fireworks_thread)
thread.start()
time.sleep(5)
thread.stop()
time.sleep(5)

imageio.mimsave('fireworks.gif', images)