如何合并实部虚部阻抗图像数组并解决保存维度报错问题

一、调整数组维度为(50x50x2) #

你当前生成的Image形状是(2,50,50)，需要将轴的顺序调整为(50,50,2)，可通过以下两种方法实现：

方法1：使用np.transpose #

import numpy as np

# 合并实部和虚部数组
Image = np.array([tag_gadf_re[0], tag_gadf_im[0]])
# 调整轴顺序：从(2,50,50)转为(50,50,2)
Image = np.transpose(Image, (1, 2, 0))

方法2：使用np.moveaxis #

import numpy as np

Image = np.array([tag_gadf_re[0], tag_gadf_im[0]])
# 将第0轴（通道轴）移到最后一维
Image = np.moveaxis(Image, 0, -1)

二、解决图像保存报错问题 #

plt.imsave仅支持3通道（RGB）或4通道（RGBA）的常规图像格式，2通道数组无法直接保存，可按以下两种场景处理：

场景1：需要保存为可视化图像 #

给数组添加第三个通道，转为3通道格式：

import matplotlib.pyplot as plt

# 方法A：补0作为第三个通道
Image_3ch = np.concatenate([Image, np.zeros((50,50,1))], axis=-1)
# 方法B：复制其中一个通道作为第三个通道（比如实部通道）
Image_3ch = np.concatenate([Image, Image[...,:1]], axis=-1)

# 保存图像
plt.imsave("your_directory/image.png", Image_3ch)

场景2：仅需保存数据（无需可视化） #

直接用numpy保存数组文件，保留原始2通道结构：

# 保存为npy文件
np.save("your_directory/image_data.npy", Image)

# 后续加载数据
loaded_image = np.load("your_directory/image_data.npy")

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