Python 如何使用透明像素将遮罩数组保存为png？

我有一个屏蔽的numpy数组（例如2000x2000，dtype=np.uint16）

我想将其存储为带有alpha通道的灰度png文件。像素应该具有元素的值，但对于遮罩元素是透明的

最好的方法是什么

---

谢谢你的建议

你是说这些蒙版元素应该是透明的？因为每个像素都包含alpha通道！当然可以。为了清楚起见，我将编辑这个问题。我认为matplotlib函数

imsave

可以自动将matplotlib保存为8位像素。然而，我发现numpngw在5分钟内解决了这个问题。这是一个非常友好的包。你是说这些蒙版元素应该是透明的？因为每个像素都包含alpha通道！当然可以。为了清楚起见，我将编辑这个问题。我认为matplotlib函数

imsave

可以自动将matplotlib保存为8位像素。然而，我发现numpngw在5分钟内解决了这个问题。这是一个非常友好的包。