用Python和Matplotlib搞定高光谱图像可视化：从.mat文件到伪彩色图（附完整代码）

PythonMatplotlib高光谱图像可视化实战从.mat文件到伪彩色合成高光谱图像处理正逐渐从专业遥感领域走向更广泛的工业应用场景。当一位农业科技公司的算法工程师第一次拿到作物生长监测的高光谱数据时面对.mat格式文件中那个神秘的三维矩阵最迫切的需求往往不是复杂的分析算法而是最基础的——让我先看看这些数据长什么样。本文将手把手带您实现高光谱数据的可视化突破解决这个数据在手却看不见的典型痛点。1. 环境准备与数据加载在开始可视化之旅前需要确保工作环境配置正确。推荐使用Anaconda创建专属的Python环境conda create -n hyperspectral python3.8 conda activate hyperspectral pip install numpy scipy matplotlib pillow高光谱数据通常以.mat格式存储这是MATLAB的标准数据格式。Python中通过scipy.io模块可以无缝读取import scipy.io as sio import matplotlib.pyplot as plt # 加载.mat文件 data_path Indian_pines.mat # 示例数据集 data sio.loadmat(data_path) hsi_cube data[indian_pines] # 获取三维数据矩阵 print(f数据维度{hsi_cube.shape}) # 典型输出(145, 145, 200)注意不同数据集中的主键名称可能不同常见的有cube、data、img等需根据实际情况调整。数据集通常包含以下维度前两维空间分辨率高度×宽度第三维光谱波段数为方便后续处理建议立即进行数据归一化hsi_cube (hsi_cube - hsi_cube.min()) / (hsi_cube.max() - hsi_cube.min())2. 单波段图像可视化高光谱数据的每个波段都相当于一个灰度图像展示特定波长下的反射率。查看单个波段是理解数据的第一步def show_single_band(cube, band_idx, titleNone): plt.figure(figsize(8, 6)) plt.imshow(cube[:, :, band_idx], cmapgray) plt.colorbar(label反射率强度) plt.axis(off) if title: plt.title(f波段 {band_idx} (约{int(band_idx*5400)}nm)) plt.show() show_single_band(hsi_cube, 50) # 查看第50个波段典型问题排查表问题现象可能原因解决方案图像全白数据未归一化执行Min-Max归一化图像全黑波段索引超出范围检查cube.shape[2]颜色异常使用了彩色映射设置cmapgray进阶技巧创建波段浏览工具动态查看不同波段from ipywidgets import interact interact(band(0, hsi_cube.shape[2]-1)) def explore_bands(band0): show_single_band(hsi_cube, band)3. 伪彩色合成技术人眼只能感知RGB三原色伪彩色合成是将非可见光波段映射到可见光范围的艺术。关键在于波段选择策略3.1 基础三波段合成def rgb_composite(cube, bands): 三波段伪彩色合成 bands: 包含R,G,B三个波段索引的列表如[29, 19, 9] rgb cube[:, :, bands] # 对各通道单独归一化以增强对比度 for i in range(3): rgb[:,:,i] (rgb[:,:,i] - rgb[:,:,i].min()) / (rgb[:,:,i].max() - rgb[:,:,i].min()) plt.figure(figsize(10, 8)) plt.imshow(rgb) plt.title(f伪彩色合成 (R:{bands[0]}, G:{bands[1]}, B:{bands[2]})) plt.axis(off) plt.show() # 示例植被分析常用波段组合 rgb_composite(hsi_cube, [50, 30, 20]) # 近红外、红边、绿波段3.2 智能波段选择算法手动选择波段依赖经验以下算法可自动寻找最佳展示波段def optimal_band_selection(cube, n_bands3): 基于波段方差的自动选择 band_vars np.var(cube, axis(0,1)) sorted_bands np.argsort(band_vars)[-n_bands:] return sorted_bands[::-1] # 按方差降序排列 best_bands optimal_band_selection(hsi_cube) print(f推荐波段组合{best_bands}) rgb_composite(hsi_cube, best_bands)不同应用场景的波段选择建议应用领域推荐波段组合特征提取植被监测[50,30,20]增强叶绿素差异水体检测[15,10,5]突出水体吸收特征矿物识别[100,80,60]增强矿物光谱特征4. 高级可视化技巧4.1 光谱曲线提取通过点击图像查看单个像素的光谱特征def onclick(event): if event.inaxes: x, y int(event.xdata), int(event.ydata) spectrum hsi_cube[y, x, :] plt.figure() plt.plot(spectrum) plt.title(f像素({x},{y})的光谱曲线) plt.xlabel(波段索引) plt.ylabel(反射率) plt.show() plt.figure() plt.imshow(hsi_cube[:,:,50], cmapgray) plt.connect(button_press_event, onclick) plt.show()4.2 波段统计可视化# 计算各波段均值 band_means np.mean(hsi_cube, axis(0,1)) plt.figure(figsize(12,4)) plt.plot(band_means) plt.fill_between(range(len(band_means)), band_means - np.std(hsi_cube, axis(0,1)), band_means np.std(hsi_cube, axis(0,1)), alpha0.3) plt.title(各波段反射率统计) plt.xlabel(波段索引) plt.ylabel(平均反射率) plt.grid() plt.show()4.3 交互式三维可视化from mpl_toolkits.mplot3d import Axes3D def plot_3d_spectral(x, y): fig plt.figure(figsize(10,6)) ax fig.add_subplot(111, projection3d) # 绘制光谱曲线 z range(hsi_cube.shape[2]) ax.plot(z, [y]*len(z), hsi_cube[x,y,:], r-, linewidth2) # 绘制空间位置 ax.scatter([0], [y], [hsi_cube[x,y,0]], cb, s100) ax.set_xlabel(波段维度) ax.set_ylabel(空间Y轴) ax.set_zlabel(反射率) plt.title(f空间位置({x},{y})的光谱特征) plt.show() plot_3d_spectral(80, 60) # 示例坐标5. 实战案例植被健康监测以农业应用为例展示如何通过波段运算增强特定特征# 计算NDVI归一化植被指数 red_band 30 # 红波段 nir_band 50 # 近红外波段 ndvi (hsi_cube[:,:,nir_band] - hsi_cube[:,:,red_band]) / \ (hsi_cube[:,:,nir_band] hsi_cube[:,:,red_band] 1e-10) plt.figure(figsize(10,8)) plt.imshow(ndvi, cmapRdYlGn, vmin-1, vmax1) plt.colorbar(labelNDVI值) plt.title(NDVI植被指数图) plt.axis(off) plt.show()常见指数计算表指数名称公式应用场景NDVI(NIR-Red)/(NIRRed)植被健康度NDWI(Green-NIR)/(GreenNIR)水体检测SAVI(NIR-Red)/(NIRRedL)*(1L)土壤调节植被指数在处理实际项目数据时发现波段索引与真实波长的对应关系至关重要。某次作物病害检测项目中通过对比健康与病变叶片的光谱曲线差异最终确定710nm附近的红边波段最能体现早期病害特征这成为我们监测模型的关键输入。