使用 Python 绘制粉色浪漫的爱心动图

木觞清

于 2024-05-19 16:21:30 发布

阅读量691

点赞数 11

文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_43580271/article/details/139044410

版权

在这篇博客中，我将展示如何使用 Python 绘制一个粉色浪漫的爱心动图。我们将使用 Matplotlib 库进行绘图，并使用 imageio 库将绘制的帧合成为一个动图。

第一步：安装必要的库

首先，我们需要安装所需的 Python 库。打开命令行或终端，并运行以下命令：

pip install matplotlib imageio

第二步：绘制心形曲线

我们将使用参数方程来绘制心形曲线。以下是定义心形曲线方程的代码：

import numpy as np

# 定义心形曲线方程
def heart_curve(t):
    x = 16 * np.sin(t)**3
    y = 13 * np.cos(t) - 5 * np.cos(2*t) - 2 * np.cos(3*t) - np.cos(4*t)
    return x, y

# 生成心形曲线的 t 值
t = np.linspace(0, 2*np.pi, 1000)
x, y = heart_curve(t)

在这个代码片段中，我们使用 NumPy 库生成从 0 到 2π 的一系列值，并计算出心形曲线的 x 和 y 坐标。

第三步：创建动图帧

我们将使用 Matplotlib 库绘制多个帧，并通过缩放心形曲线来实现心跳效果。以下是生成动图帧的代码：

import matplotlib.pyplot as plt
import imageio

# 创建动图帧
frames = []
for i in range(50):
    fig, ax = plt.subplots()
    ax.set_aspect('equal')
    ax.axis('off')
    ax.set_xlim(-20, 20)
    ax.set_ylim(-20, 20)
    
    # 绘制心形曲线
    x, y = heart_curve(t)
    ax.fill(x, y, 'pink', edgecolor='pink')
    
    # 添加一些动画效果
    scale = 1 + 0.05 * np.sin(i / 5 * np.pi)
    ax.fill(scale * x, scale * y, 'pink', edgecolor='pink')
    
    # 保存当前帧
    fig.canvas.draw()
    image = np.frombuffer(fig.canvas.tostring_rgb(), dtype='uint8')
    image = image.reshape(fig.canvas.get_width_height()[::-1] + (3,))
    frames.append(image)
    
    # 关闭当前帧，防止内存泄漏
    plt.close(fig)

在这个代码片段中，我们使用 for 循环生成 50 个帧。在每个帧中，我们绘制粉色填充的心形，并通过缩放心形曲线实现心跳效果。每个帧都被保存为图像并添加到 frames 列表中。

第四步：保存动图

最后，我们使用 imageio 库将帧保存为一个动图文件。以下是保存动图的代码：

# 保存为动图
imageio.mimsave('heart_animation.gif', frames, fps=10)

print("动图已保存为 heart_animation.gif")

在这个代码片段中，我们使用 imageio.mimsave 函数将 frames 列表保存为一个名为 heart_animation.gif 的动图文件。

运行代码

将以上所有代码合并到一个文件中并运行。运行后，你将得到一个名为 heart_animation.gif 的动图文件，展示了一个粉色浪漫的爱心跳动效果。

import numpy as np
import matplotlib.pyplot as plt
import imageio

# 定义心形曲线方程
def heart_curve(t):
    x = 16 * np.sin(t)**3
    y = 13 * np.cos(t) - 5 * np.cos(2*t) - 2 * np.cos(3*t) - np.cos(4*t)
    return x, y

# 生成心形曲线的 t 值
t = np.linspace(0, 2*np.pi, 1000)

# 创建动图帧
frames = []
for i in range(50):
    fig, ax = plt.subplots()
    ax.set_aspect('equal')
    ax.axis('off')
    ax.set_xlim(-20, 20)
    ax.set_ylim(-20, 20)
    
    # 绘制心形曲线
    x, y = heart_curve(t)
    ax.fill(x, y, 'pink', edgecolor='pink')
    
    # 添加一些动画效果
    scale = 1 + 0.05 * np.sin(i / 5 * np.pi)
    ax.fill(scale * x, scale * y, 'pink', edgecolor='pink')
    
    # 保存当前帧
    fig.canvas.draw()
    image = np.frombuffer(fig.canvas.tostring_rgb(), dtype='uint8')
    image = image.reshape(fig.canvas.get_width_height()[::-1] + (3,))
    frames.append(image)
    
    # 关闭当前帧，防止内存泄漏
    plt.close(fig)

# 保存为动图
imageio.mimsave('heart_animation.gif', frames, fps=10)

print("动图已保存为 heart_animation.gif")