Python 如何将 MatPlotLib 图保存为具有十进制格式的 PNG 字节数组

发布于05月14日

我有一些用MatPlotLib图形可视化的数据.我希望能够将数字转换为使用PNG十进制值格式化的字节array. https://www.w3.org/TR/PNG-Structure.html个

fig = plt.figure(figsize=(16,16))
ax = fig.add_subplot(projection='3d')
ax.voxels(space, facecolors = rgb_input)
plt.axis('off')
fig.patch.set_alpha(0.25)
ax.patch.set_alpha(0.25)

我知道如何将其保存为PNG并将其转换为字节数组，但我希望能够直接生成字节array.

我try 使用它直接获取字节数组，但它不返回PNG十进制值:

buf = io.BytesIO()
plt.savefig(buf, format = 'png')
buf.seek(0)
data = buf.read()

编辑:经过进一步研究，我发现data值被格式化为ASCII字节array.有没有办法把它转换成十进制字节数组？

推荐答案

PNG图像是二进制文件，通常以ASCII或十六进制表示，要将字节数组转换为十进制表示，只需使用Python的内置函数即可.

import io
import matplotlib.pyplot as plt
import numpy as np

space = np.random.rand(5,5,5) > 0.5
rgb_input = np.random.rand(5,5,5,3)

fig = plt.figure(figsize=(16,16))
ax = fig.add_subplot(projection='3d')
ax.voxels(space, facecolors = rgb_input)
plt.axis('off')
fig.patch.set_alpha(0.25)
ax.patch.set_alpha(0.25)

buf = io.BytesIO()
plt.savefig(buf, format = 'png')
buf.seek(0)
data = buf.read()

decimal_values = [byte for byte in data]
print(decimal_values)