問(wèn)題:
由于自己做項(xiàng)目的時(shí)候,需要循環(huán)的繪制數(shù)據(jù),假設(shè)有100個(gè)樣本,每個(gè)樣本包含兩個(gè)坐標(biāo)點(diǎn)(A, B),我需要對(duì)這兩個(gè)點(diǎn)標(biāo)上不同的顏色,同時(shí)還要畫(huà)出兩點(diǎn)間的連線。
顯然這個(gè)問(wèn)題中圖例我只需要3個(gè)(A點(diǎn),B點(diǎn),AB的連線),而不是300個(gè),因?yàn)槊總€(gè)樣本的A點(diǎn)都是同樣的顏色,B點(diǎn)也都是一樣的顏色,AB的連線也是。
但是單純的在畫(huà)完圖之后用plt.legend(), 它會(huì)給你畫(huà)出所有300個(gè)圖例來(lái),這肯定不是我想要的。
探索過(guò)程:
如何解決呢?
當(dāng)然有一種很強(qiáng)制的方法,就是只在畫(huà)第一個(gè)樣本,或最后一個(gè)樣本時(shí),給其指定label,其它樣本時(shí)不給label,就不會(huì)畫(huà)出來(lái)了。
但是我在想有沒(méi)有更好地辦法,百度了半天,窮盡所有關(guān)鍵詞,無(wú)果。
轉(zhuǎn)而谷歌,想著要不把關(guān)鍵詞換成英語(yǔ)看看(以前總覺(jué)得看英文的博客會(huì)比中文的慢很多,總會(huì)更偏向于母語(yǔ)閱讀,但是這次經(jīng)歷大概會(huì)改變我之后的態(tài)度,相比起為了排除百度的大量無(wú)關(guān)答案所花的漫長(zhǎng)的時(shí)間,好像還不如直接搜索英文來(lái)得有效率,關(guān)鍵詞搜的就是:python omit redundant legend),結(jié)果第一條搜索結(jié)果就看到了我想要的解決方案,很是驚喜,本著方便自己以后查找,也方便遇到同樣問(wèn)題的人,故而記錄下來(lái),雖然不是什么多神的東西啦。
解決方案:
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變量中,你之前所在畫(huà)布上畫(huà)的每一步應(yīng)該都會(huì)存在plt中。
所以,上述代碼的作用是:僅保留plt中存在的所有非同名的圖例名稱(chēng),不出現(xiàn)重復(fù)圖例名稱(chēng)
補(bǔ)充:Python/matplotlib之清空繪圖
clf()
# 清圖
cla()
# 清坐標(biāo)軸
close()
# 關(guān)窗口
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
您可能感興趣的文章:- Python matplotlib可視化繪圖詳解
- Python繪圖之詳解matplotlib
- Python繪圖庫(kù)Matplotlib的基本用法
- Python matplotlib實(shí)用繪圖技巧匯總
- python 繪圖模塊matplotlib的使用簡(jiǎn)介
- Python matplotlib繪圖詳解