从Python创建PNG文件并确保某些单元格透明

Question

我正在使用Python通过2D numpy数组在网格上创建各种地图。对于道路，值为1；对于背景，值为0。

我可以使用以下方式将文件导出为PNG文件：

matplotlib.image.imsave（filename.png，map_array）

PNG文件已创建，例如，我可以看到所有道路都使用一种颜色（黄色），背景看到另一种颜色（紫色）。但是，我希望将所有1（道路）都引用为特定的颜色（例如红色），而背景要是透明的，因为我将地图覆盖在地图上。

如何使PNG中的零值完全透明，并为PNG的1赋予特定颜色。

仅供参考，我试图将零值更改为null（np.NaN），但它似乎不起作用，只是使所有内容变为紫色。

Answer 1

尝试在transparent=True函数中使用参数.imsave