python matplotlib繪圖實(shí)現(xiàn)刪除重復(fù)冗余圖例的操作

問題：

由于自己做項(xiàng)目的時(shí)候，需要循環(huán)的繪制數(shù)據(jù)，假設(shè)有100個(gè)樣本,每個(gè)樣本包含兩個(gè)坐標(biāo)點(diǎn)（A, B），我需要對這兩個(gè)點(diǎn)標(biāo)上不同的顏色，同時(shí)還要畫出兩點(diǎn)間的連線。

顯然這個(gè)問題中圖例我只需要3個(gè)（A點(diǎn)，B點(diǎn)，AB的連線），而不是300個(gè)，因?yàn)槊總€(gè)樣本的A點(diǎn)都是同樣的顏色，B點(diǎn)也都是一樣的顏色，AB的連線也是。

但是單純的在畫完圖之后用plt.legend(), 它會給你畫出所有300個(gè)圖例來，這肯定不是我想要的。

探索過程：

如何解決呢？

當(dāng)然有一種很強(qiáng)制的方法，就是只在畫第一個(gè)樣本，或最后一個(gè)樣本時(shí)，給其指定label，其它樣本時(shí)不給label，就不會畫出來了。

但是我在想有沒有更好地辦法，百度了半天，窮盡所有關(guān)鍵詞，無果。

轉(zhuǎn)而谷歌，想著要不把關(guān)鍵詞換成英語看看（以前總覺得看英文的博客會比中文的慢很多，總會更偏向于母語閱讀，但是這次經(jīng)歷大概會改變我之后的態(tài)度，相比起為了排除百度的大量無關(guān)答案所花的漫長的時(shí)間，好像還不如直接搜索英文來得有效率，關(guān)鍵詞搜的就是：python omit redundant legend），結(jié)果第一條搜索結(jié)果就看到了我想要的解決方案，很是驚喜，本著方便自己以后查找，也方便遇到同樣問題的人，故而記錄下來，雖然不是什么多神的東西啦。

解決方案：

from collections import OrderedDict
import matplotlib.pyplot as plt
 
handles, labels = plt.gca().get_legend_handles_labels()
by_label = OrderedDict(zip(labels, handles))
plt.legend(by_label.values(), by_label.keys())

理解：

plt的legend函數(shù)可接收兩個(gè)參數(shù)，一個(gè)是handles列表，一個(gè)是label列表。

引入有序字典OrderedDict的目的就是去除冗余的圖例，因?yàn)樽值涞逆I值不能重復(fù)（即重復(fù)的只保留一個(gè)）。

第一行的意思應(yīng)該是迭代地將當(dāng)前plt中存的所有handles和labels返回到handles變量和labels變量中，你之前所在畫布上畫的每一步應(yīng)該都會存在plt中。

所以，上述代碼的作用是：僅保留plt中存在的所有非同名的圖例名稱，不出現(xiàn)重復(fù)圖例名稱

補(bǔ)充：Python/matplotlib之清空繪圖

clf() # 清圖

cla() # 清坐標(biāo)軸

close() # 關(guān)窗口

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。