爲每個數據點繪製不同顏色的動畫圖

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()