Matplotlib imsave报错：ndarray非C连续（实际为连续状态？）

问题描述 #

我想用Matplotlib的imsave()函数，将包含单色图像数据的2D numpy数组（含需显示为透明的np.nan值）结合viridis颜色映射保存为PNG文件。由于imsave无法直接处理np.nan，我先将数组转换为RGBA格式。使用默认的origin='upper'时一切正常，但改为origin='lower'时，Matplotlib抛出“ndarray is not C-contiguous”错误（实际来自底层PIL库）。

完整示例代码 #

import numpy as np
from matplotlib import cm
import matplotlib.pyplot as plt

# 创建测试数据
image_no_nan = np.eye(100)
image = np.copy(image_no_nan)
image[image == 0] = np.nan

# 将图像映射到RGBA，缩放至[0-255]范围
colmap = plt.get_cmap("viridis", 256)
lut = (colmap.colors[..., 0:4] * 255).astype(np.uint8)
rescaled = (
    (image_no_nan.astype(float) - image_no_nan.min())
    * 255
    / (image_no_nan.max() - image_no_nan.min())
).astype(np.uint8)

result = np.zeros((*rescaled.shape, 4), dtype=np.uint8)

# 根据灰度值从LUT中提取对应RGBA值
result = np.take(lut, rescaled, axis=0, out=result)

# 应用透明掩码
mask = np.zeros((rescaled.shape), dtype=np.uint8)
mask[~np.isnan(image)] = 255
result[:,:,3]= mask

# 尝试修复C-contiguous错误，但无效
result = result.copy(order="C")
result = np.ascontiguousarray(result)

print(result.flags) # 显示数组确实是C-contiguous

plt.imsave(fname="test_upper.png", arr=result, format="png", origin="upper") # 无问题
plt.imsave(fname="test_lower.png", arr=result, format="png", origin="lower") # 报错

问题根源 #

当设置origin='lower'时，Matplotlib会对输入的RGBA数组执行垂直翻转操作，但这个操作只会生成原数组的一个非C-contiguous视图（不会创建新的连续数组）。而底层的PIL库要求输入的数组必须是C-contiguous的，因此触发了错误。

你手动执行的copy(order="C")或np.ascontiguousarray仅处理了原始数组，但翻转后的视图依然是非连续的，所以无法解决问题。

解决方案 #

方法1：手动翻转并强制转为连续数组 #

在传递给imsave前，手动对数组进行垂直翻转，并确保生成的数组是C-contiguous的，之后使用origin='upper'保存：

# 手动垂直翻转并转为连续数组
result_lower = np.ascontiguousarray(np.flipud(result))
plt.imsave(fname="test_lower.png", arr=result_lower, format="png", origin="upper")

方法2：直接使用PIL库保存 #

绕开Matplotlib的翻转逻辑，直接用PIL处理数组并保存：

from PIL import Image

# PIL默认origin为upper，要实现lower效果需先垂直翻转
img = Image.fromarray(np.flipud(result))
img.save("test_lower.png")

方法3：将数组转为PIL Image后传入imsave #

imsave支持直接传入PIL Image对象，此时Matplotlib不会再对数组进行翻转操作：

from PIL import Image

img = Image.fromarray(result)
plt.imsave(fname="test_lower.png", arr=img, format="png", origin="lower")

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