Matplotlib：改變索引中特定點之後的行的顏色

Question

我一直在探索Matplotlib和熊貓的一個月的升技，我想知道是否有任何方法可以改變給定圖形的行的顏色在索引的某個值通過後，我在Matplotlib文檔中找不到任何類似的東西，並且非常感謝任何人的專業知識。

ax1 = plt.subplot2grid((1,1), (0,0)) 

num = 0 
for i in df['GDP']: 
    num +=1 
    if num > 263: 
     df.plot(color = 'r', ax=ax1) 
    else: 
     df.plot(color= 'r', ax=ax1) 
plt.show() 

我以爲這樣的事情可以工作，但對我的運氣它沒有。

注：IM與GDP數據是在我的數據幀的唯一列，如果有人想知道:)

Answer 1

您可以通過boolean indexing與loc覆蓋行選擇列GDP，其中比較指數值工作（index有被單調遞增，默認）：

np.random.seed(100) 
df = pd.DataFrame(np.random.randint(10, size=(30,2)), columns=['GDP', 'A']) 
#print (df) 

N = 20 
ax = df['GDP'].plot(color='y') 
df.loc[df.index >= N, 'GDP'].plot(color='r', ax=ax) 
plt.show()