Matplotlib动画实现新点红色、旧点灰色的正弦散点图问题求助

你好呀！我来帮你看看这段代码的问题～你的需求是做一个正弦散点动画，让新出现的点是红色、已经出现的旧点变成灰色，这个思路很有意思，不过你的代码里有几个小问题导致达不到预期效果：

• 每次调用ax.plot都会新增绘图元素：你在animate里每次都重新调用ax.plot(x1[:i+1], y1[:i+1], '*r')，这会在画布上不断叠加新的散点图层，而不是更新已有图层。之前的graph.set_color('gray')其实是把上一次的整个散点图层设为灰色，但新的图层又包含了所有点，最后画面会变得混乱，颜色逻辑也不对。
• init函数的返回值有问题：初始时graph是None，init函数返回graph,会导致动画初始化报错，因为没有有效的绘图对象。
• 全局变量的管理容易出错：用global graph来跟踪绘图对象不是最优方式，而且逻辑上没有区分“旧点”和“新点”，只是把整个图层变灰再重新画所有点，这和你想要的“逐个点变灰”的效果不符。

那我们来调整一下代码，换个思路：维护两个散点对象，一个专门放灰色的旧点，一个放红色的新点。每次动画帧更新时，把上一个红色点移到旧点集合，然后添加新的红色点，这样就能精准控制每个点的颜色啦：

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

x1 = np.linspace(0, 10, 1000)
y1 = np.sin(2 * np.pi * x1)

fig, ax = plt.subplots()
ax.set_xlim([0, 5])
ax.set_ylim([-2, 2])

# 初始化两个散点对象：灰色旧点、红色新点
old_points, = ax.plot([], [], '*', color='gray')
new_point, = ax.plot([], [], '*', color='red')

def init():
    # 初始状态下没有点，返回两个散点对象
    old_points.set_data([], [])
    new_point.set_data([], [])
    return old_points, new_point

def animate(i):
    # 获取当前要添加的点坐标
    current_x = x1[i]
    current_y = y1[i]
    
    # 获取旧点已有的数据，把之前的新点加进去
    old_x, old_y = old_points.get_data()
    if i > 0:
        # 从第2帧开始，把上一个点移到旧点集合
        old_x = np.append(old_x, x1[i-1])
        old_y = np.append(old_y, y1[i-1])
        old_points.set_data(old_x, old_y)
    
    # 更新新点为当前点
    new_point.set_data(current_x, current_y)
    
    return old_points, new_point

# 设置动画，frames设为你想要的帧数，这里用前50个点为例
ani = animation.FuncAnimation(fig, animate, init_func=init, frames=50, interval=50)
plt.show()

这段代码的逻辑很清晰：

1. 一开始创建两个plot对象，分别对应灰色旧点和红色新点，避免每次创建新的绘图元素，而是直接更新它们的数据。
2. init函数负责把两个散点对象初始化为空数据，确保动画开始时画布干净。
3. 在animate函数里：
   • 先拿到当前帧对应的点坐标
   • 如果不是第一帧，就把上一个点添加到旧点集合里，更新旧点的数据
   • 把新点设置为当前帧的点，这样每次只有最新的点是红色，之前的都在灰色集合里

你可以根据需求调整frames参数的数值，控制动画展示的点数~

备注：内容来源于stack exchange，提问作者aRedDish