前言
遙感影像分類圖一般為特定數(shù)值對應(yīng)一類地物，用Python繪制時，主要在顏色的映射和對應(yīng)的圖例生成。
plt.matplotlib.colors.ListedColormap支持自定義顏色。matplotlib.patches mpatches對象可以生成一個矩形對象，控制其顏色和地物類型的顏色對應(yīng)就可以生成地物分類的圖例了。具體用法可以自行Google和百度。下面給出一個模擬地物分類數(shù)據(jù)的可視化例子。
代碼
import numpy as np
import matplotlib.pyplot as plt
np.random.seed(0)
data = np.random.randint(0, 3, size=(100,100))
colors = dict((
(0, (0, 255, 0, 255)), # 前三位RGB，255代表256色
(1, (0, 0, 255, 255)),
(2, (255, 255, 0, 255)),
))
# 轉(zhuǎn)換為0-1
for k in colors:
v = colors[k]
_v = [_v / 255.0 for _v in v]
colors[k] = _v
index_colors = [colors[key] if key in colors else
(255, 255, 255, 0) for key in range(0, len(colors))]
cmap = plt.matplotlib.colors.ListedColormap(index_colors, 'Classification', len(index_colors)) # n等于顏色表長度，否則被截斷或被重復(fù)
# cmap = plt.matplotlib.colors.ListedColormap(['gray', 'orange', 'k'], 'Classification')
plt.rcParams['font.family'] = 'Arial'
plt.rcParams['font.size'] = 10
plt.rcParams['font.weight'] = 'bold'
fig, ax = plt.subplots(figsize=(4,3.5), dpi=300)
ax.imshow(data, cmap=cmap, interpolation='none')
# 繪制矩形的補丁, 用來生成圖例，fig.add_artist()才會在圖中顯示出來
import matplotlib.patches as mpatches
rectangles = [mpatches.Rectangle((0, 0,), 1, 1, facecolor=index_colors[i])
for i in range(len(index_colors))]
labels = ['forest',
'water',
'urban']
ax.legend(rectangles, labels,
bbox_to_anchor=(1.4, 0.25), fancybox=True, frameon=False,)
# 取消刻度和標(biāo)簽顯示
ax.tick_params(which='major', bottom=0, left=0)
ax.set_xticklabels('')
ax.set_yticklabels('')
效果圖：
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
標(biāo)簽：赤峰 金華 陽泉 臨汾 雙鴨山 貴州 克拉瑪依 日照
巨人網(wǎng)絡(luò)通訊聲明：本文標(biāo)題《Python繪制分類圖的方法》，本文關(guān)鍵詞 Python,繪制,分類,圖,的,方法,；如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題，煩請?zhí)峁┫嚓P(guān)信息告之我們，我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò)，涉及言論、版權(quán)與本站無關(guān)。
