用python写一个跳动的爱心代码

以下是用Python实现跳动的爱心代码：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# 设置画布大小
fig, ax = plt.subplots(figsize=(8, 6))

# 设置坐标轴范围
ax.set_xlim(-2, 2)
ax.set_ylim(-2, 2)

# 定义爱心函数
def heart(t):
    x = np.sin(t)
    y = np.cos(t) + np.power(np.abs(x), 0.5)
    return x, y

# 初始化函数
def init():
    line.set_data([], [])
    return line,

# 更新函数
def update(frame):
    x, y = heart(frame)
    line.set_data(x, y)
    return line,

# 绘制爱心
line, = ax.plot([], [], lw=2)
ani = FuncAnimation(fig, update, frames=np.linspace(0, 2*np.pi, 100),
                    init_func=init, blit=True)

# 保存动画
ani.save('heart.gif', writer='imagemagick', fps=30)