如何将Plots.jl绘制的Julia图像转换为Matrix{RGB{N0f8}}

完全可以实现类似Matplotlib的转换逻辑，以下是完整解决方案：

步骤说明 #

1. 用IOBuffer在内存中存储Plots.jl生成的原始图像字节流
2. 读取字节数据并转换为UInt8数组
3. 调整数组维度匹配图像尺寸
4. 将RGBA数据转换为Matrix{RGB{N0f8}}类型

完整代码 #

using Plots, ColorTypes, FixedPointNumbers

# 生成目标热力图
n = 255
I = [i for i in 0:n, j in 0:n]
h = heatmap(mod.((I .| I'), 19), c=:deep, yflip=true, size=(400, 400), aspect_ratio=:equal)

# 内存中保存原始图像数据
io_buf = IOBuffer()
savefig(h, io_buf, format="raw")
seekstart(io_buf)

# 读取字节流为UInt8数组（对应numpy.frombuffer的功能）
raw_bytes = read(io_buf, UInt8)

# 获取图像尺寸并重塑数组：raw格式为RGBA四通道，按(height, width, 4)排列
width, height = size(h)
rgba_array = reshape(raw_bytes, (4, width, height)) |> permutedims([3, 2, 1])

# 转换为Matrix{RGB{N0f8}}：提取RGB通道并转换为N0f8固定点类型
rgb_matrix = RGB{N0f8}.(rgba_array[:, :, 1]./255, rgba_array[:, :, 2]./255, rgba_array[:, :, 3]./255)

# 验证结果
println("转换后的类型：", typeof(rgb_matrix))
println("转换后的尺寸：", size(rgb_matrix))

关键细节说明 #

• IOBuffer与savefig：Julia的IOBuffer等价于Python的io.BytesIO，配合savefig(..., format="raw")可以输出无压缩的RGBA原始字节流，和Matplotlib的逻辑一致
• 字节读取与数组重塑：read(io_buf, UInt8)实现了类似numpy.frombuffer的字节解析功能，将内存中的字节流转换为UInt8数组；通过reshape和permutedims调整维度顺序，匹配图像的高、宽、通道结构
• RGB{N0f8}转换：N0f8是FixedPointNumbers包提供的固定点数类型，将0-255的UInt8值映射为0-1的浮点数范围，直接通过除法后转换即可得到目标类型

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