©2020-2024 ioDraw All rights reserved,
Privacy Policy
手把手教你用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)