如何在matplotlib中更改动画线条的颜色？

import matplotlib.pyplot as plt
import matplotlib.animation as animation

fig, ax = plt.subplots()

line, = ax.plot([], [], lw=2)

def init():
    line.set_data([], [])
    line.set_color('blue')  # 设置线条颜色为蓝色
    return line,

def update(frame):
    x = [0, 1, 2, 3, 4]  # 示例数据
    y = [0, 1, 4, 9, 16]  # 示例数据
    line.set_data(x[:frame], y[:frame])  # 更新线条的数据
    line.set_color('red')  # 设置线条颜色为红色
    return line,

ani = animation.FuncAnimation(fig, update, frames=len(x), init_func=init, blit=True)

plt.show()