在matplotlib中加入NaN值

Question

我有一個列表y，其中包含nan每個後面的實數，以防止線條在matplotlib中繪製時加入。我可以嘗試通過屏蔽nan數據來與np.isfinite()作圖。但是，當它們大於5時，我可以忽略nan - 因此，在超過5個nan s的區域中不會發生線連接。 pandas已經內置series.fillna(limit=2)這似乎很理想，但我不想取代nan。我真的需要一個例子來說明如何做到這一點。

我想要的輸出結果如下（7到14之間的差距）。

import matplotlib.pyplot as plt 
import numpy as np 
x = np.arange(20) 
y = [np.nan, 5, np.nan, np.nan, 3, np.nan, 10, np.nan, np.nan, np.nan, np.nan, np.nan, np.nan, 2, np.nan, 7, np.nan, 22, np.nan, 15, np.nan] 

plt.plot(x,y, '-') 

Answer 1

我沒有看到一個乾淨地矢量化這一點。

如果你真的只點了一把，只寫一個循環做

def brute_force_clean_nans(x, y): 
    x_clean, y_clean = [], [] 
    cnt = 0 
    for _x, _y in zip(x, y): 
     if np.isnan(_y): 
      cnt += 1 
      if cnt == 5: 
       # on the 5th nan, put it in the list to break line 
       x_clean.append(_x) 
       y_clean.append(_y) 
      continue 
     cnt = 0 
     x_clean.append(_x) 
     y_clean.append(_y) 
    return x_clean, y_clean 

你也可以做的事情有np.where，看着奔跑等，但如果你有這幾點它可能不值得。