手把手教你用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)