如何保存与原数组尺寸一致的颜色映射数组？

我完全理解你的需求——要让保存的PNG严格对应原始数据的每个像素，不能有任何插值、缩放，同时还要用上matplotlib的颜色映射。直接用savefig确实会踩坑，因为它默认保存的是整个绘图画布（包括轴、边距），还会根据dpi缩放，导致像素被模糊处理。下面给你两种最实用的实现方案，都能完美解决你的问题：

方案一：精准控制画布尺寸，提取纯图像保存 #

这种方法适合你需要先在交互式窗口查看图像，再保存精确像素版本的场景。核心思路是让画布大小完全匹配数据的像素尺寸，去掉所有多余元素（轴、边距），确保每个数据点对应一个图像像素。

修改你的示例代码如下：

import numpy as np
import matplotlib.pyplot as plt

# 生成原始数据
X,Y = np.meshgrid(np.arange(-50.0,50,.1), np.arange(-50.0,50,.1))
Z = np.abs(np.sin(2*np.pi*(X**2+Y**2)**.5))/(1+(X/20)**2+(Y/20)**2)

# 获取数据的像素维度（行=高度，列=宽度）
height, width = Z.shape

# 设定dpi，这里选100是常规值，你也可以选其他，只要保持前后一致
dpi = 100
# 计算画布尺寸：像素数 / dpi = 英寸数，确保画布刚好容纳所有像素
fig = plt.figure(figsize=(width/dpi, height/dpi), dpi=dpi)
# 添加轴并让它填满整个画布，彻底去掉边距
ax = fig.add_axes([0, 0, 1, 1])

# 绘制图像，interpolation='nearest'强制关闭插值，保证像素锐利
ax.imshow(Z, cmap='inferno', interpolation='nearest')

# 关闭坐标轴显示（我们只需要纯图像）
ax.axis('off')

# 保存时关闭边距，确保没有多余空白
plt.savefig('colormapeg_exact.png', dpi=dpi, bbox_inches='tight', pad_inches=0)
plt.close()

这样保存的PNG，每个像素都严格对应Z数组里的一个元素，放大后和你在交互式窗口里看到的细节完全一致，没有任何模糊。

方案二：直接生成颜色映射数组，绕过绘图环节 #

如果不需要交互式查看，只是想把数据转成带颜色映射的PNG，这个方法更高效——直接操作数组，跳过matplotlib的绘图渲染步骤。

示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 生成原始数据
X,Y = np.meshgrid(np.arange(-50.0,50,.1), np.arange(-50.0,50,.1))
Z = np.abs(np.sin(2*np.pi*(X**2+Y**2)**.5))/(1+(X/20)**2+(Y/20)**2)

# 用matplotlib自带的imsave直接保存，一步到位
# 这个函数会自动处理颜色映射，且严格按照数组尺寸生成像素，无插值
plt.imsave('colormapeg_direct.png', Z, cmap='inferno', vmin=Z.min(), vmax=Z.max())

如果你想手动控制颜色映射的细节（比如归一化范围），也可以拆解步骤：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import Normalize
from PIL import Image

# 生成数据
Z = ... # 你的原始数据

# 归一化数据到[0,1]区间，适配颜色映射的输入要求
norm = Normalize(vmin=Z.min(), vmax=Z.max())
normalized_Z = norm(Z)

# 获取指定的颜色映射
cmap = plt.get_cmap('inferno')

# 应用颜色映射得到RGBA数组，转换为0-255的RGB整数数组
rgba_array = cmap(normalized_Z)
rgb_array = (rgba_array[..., :3] * 255).astype(np.uint8)

# 用PIL保存为PNG
img = Image.fromarray(rgb_array)
img.save('colormapeg_manual.png')

为什么这两种方法有效？ #

• 方案一通过精准控制画布尺寸和dpi，让每个数据点刚好对应一个图像像素，关闭插值和多余元素后，保存的就是纯原始像素的图像。
• 方案二直接把数组转成颜色映射后的RGB数据，本质上是将数值映射为颜色值，再以原始尺寸保存，完全绕开了绘图环节的缩放和插值问题。

这两种方法都能满足你的需求：严格保留原始像素尺寸，使用matplotlib的颜色映射，没有任何插值或模糊。

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