Matplotlib设置NaN值填充的区域为透明

Matplotlib将数据绘制为带有颜色的图片特别方便。
有一个需求：将生成的图片的填充值位置变为透明色。

单通道数据

import matplotlib.pyplot as plt
alpha = 0
fig = plt.figure()
fig.patch.set_alpha(alpha)
plt.savefig('alpha.png')

.html

多通道image数据

使用RGBA，A通道赋值为0是透明。