如何在MATLAB中复现Python plt.imshow功能？含vmin/vmax及bwr色图需求

一、实现vmin/vmax的颜色归一化 #

MATLAB的imagesc或imshow配合caxis可以完全复现plt.imshow的vmin/vmax逻辑：

• imagesc(X)默认会自动缩放颜色匹配X的极值，但通过caxis([vmin, vmax])可以强制限定颜色映射范围，超出范围的数值会被截断到色图的两端，和Python的vmin/vmax行为一致。
• 若使用imshow，对浮点数据可直接通过imshow(X, 'DisplayRange', [vmin, vmax])实现相同效果，无需手动归一化。

示例对应代码（以实部显示为例）：

imagesc(real(x));
caxis([-mx, mx]); % 等价于plt.imshow的vmin=-mx, vmax=mx

二、导入Python专属色图（以bwr为例） #

MATLAB无原生bwr色图，但可通过两种方式实现：

方法1：手动生成对称蓝-白-红渐变 #

bwr色图是蓝→白→红的对称渐变，直接用MATLAB内置函数生成：

% 生成256级标准bwr色图
n = 256;
blue_white = linspace(0, 1, n/2)' * [1 1 0] + [0 0 1];
white_red = linspace(1, 0, n/2)' * [1 1 0] + [0 0 1];
bwr = [blue_white; flip(white_red, 1)];
colormap(bwr);

方法2：从Matplotlib导出完全一致的色图数据 #

若需要和Python端完全匹配的色图，先在Python中导出RGB数据：

import matplotlib.pyplot as plt
import numpy as np

cmap = plt.get_cmap('bwr', 256)
colors = cmap(np.linspace(0, 1, 256))[:, :3] # 剔除alpha通道
np.save('bwr_colormap.npy', colors)

再在MATLAB中加载使用：

colors = readmatrix('bwr_colormap.npy');
colormap(colors);

注：MATLAB R2020b及以后版本已原生支持turbo色图，直接调用colormap(turbo)即可。

三、完整复现Python示例的MATLAB代码 #

%% 生成与Python等价的x数据
N = 2048;
M = 512;
wx = linspace(0, 1, N);
wy = linspace(0, 1, M);

psi_x = ifftshift(ifft(exp(-(wx - 0.0125/2).^2 * 2^19)));
psi_y = ifftshift(ifft(exp(-(wy - 0.00625*2).^2 * 2^16)));
x = psi_x .* psi_y'; % 转置匹配Python的[:, None]维度逻辑

mx = max(abs(x(:)));

%% 绘制实部（bwr色图）
figure('Position', [100, 100, 800, 800]); % 对应Python的set_size_inches(8,8)
imagesc(real(x));
colormap(bwr);
caxis([-mx, mx]);
axis('auto'); % 对应aspect='auto'
title('Real part', 'Location', 'left', 'FontWeight', 'bold', 'FontSize', 22);
colorbar;

%% 绘制绝对值（turbo色图）
figure('Position', [100, 100, 800, 800]);
imagesc(abs(x));
colormap(turbo);
caxis([0, mx]);
axis('auto');
title('Absolute value', 'Location', 'left', 'FontWeight', 'bold', 'FontSize', 22);
colorbar;

内容的提问来源于stack exchange，提问作者OverLordGoldDragon