Question

我想创建一个时间序列的动画图，但我希望能够以不同的方式为每个数据点着色。当我在时间序列数据上运行各种分析任务时，我想根据它所属的区域为每个数据点着色。

我按照example来了解动画绘图是如何工作的，我还发现answer展示了如何合并颜色。问题在于，在该方法中，整个图形在每次迭代中被重新绘制，从而改变整个图形的颜色，而不是仅改变新绘制的数据点。

有人可以告诉我如何更改衰减示例以为每个数据点指定不同的颜色吗？

Answer 1

您可以使用scatter对点进行着色，并且如果您没有计划绘制太多点，只需添加新点，每次使用不同的颜色可能是最佳选择。一个基于衰变的最小例子，

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


def data_gen(t=0):
    cnt = 0
    while cnt < 1000:
        cnt += 1
        t += 0.01
        yield t, np.sin(2*np.pi*t) * np.exp(-t/10.)

def get_colour(t):
    cmap = matplotlib.cm.get_cmap('Spectral')
    return cmap(t%1.)

def init():
    ax.set_ylim(-1.1, 1.1)
    ax.set_xlim(0, 10)


fig, ax = plt.subplots()
ax.grid()

def run(data):

    # Get some data and plot
    t, y = data
    ax.scatter(t, y, c=get_colour(t))

    #Update axis
    xmin, xmax = ax.get_xlim()
    if t >= xmax:
        ax.set_xlim(xmin, 2*xmax)
        ax.figure.canvas.draw()

ani = animation.FuncAnimation(fig, run, data_gen, blit=False, interval=10,
                              repeat=False, init_func=init)
plt.show()

每个数据点都有不同颜色的动画图

1 个答案: