Matplotlib 图像绘图 - nan 值显示为颜色图的最低颜色而不是透明

Question

我正在使用matplotlib 3.0.3 并想使用FuncAnimation 模块创建图像绘图动画。对于绘图速度，我使用im.set_data 为imshow 和im.set_array() 为plt.pcolormesh 更新图像数据（set_data 不能用作属性）。如果我用 NaN 值部分更新数据，imshow 将它们显示为空白像素，而pcolormesh 将它们显示为颜色图中的最低颜色（viridis 为蓝色）。

这是有意的，如果不是，为什么会这样？它似乎与set_array() 有关，因为pcolormesh 通常会将NaN 绘制为空白像素。

小例子：

import numpy as np
import matplotlib.pyplot as plt

fig1 = plt.figure()
im1 = plt.pcolormesh(np.random.rand(10,10))
im1.set_array((np.zeros((10,10)) * np.nan).ravel())

fig2 = plt.figure()
im2 = plt.imshow(np.random.rand(10,10))
im2.set_data(np.zeros((10,10)) * np.nan)

Answer 1

Matplotlib 图像不适用于 nans。应该改为使用掩码数组。那么这两种情况都是一样的（除了需要展平pcolormesh数组）。

import numpy as np
import matplotlib.pyplot as plt

A = np.random.rand(10,10)
B = np.ma.array(A, mask=np.ones((10,10)))

fig1 = plt.figure()
im1 = plt.pcolormesh(A)
im1.set_array(B.ravel())
plt.colorbar()

fig2 = plt.figure()
im2 = plt.imshow(A)
im2.set_array(B)
plt.colorbar()

plt.show()