Làm cách nào để vẽ các giá trị NaN dưới dạng màu đặc biệt với imshow trong matplotlib?

Tôi đang cố gắng sử dụng imshow trong matplotlib để vẽ dữ liệu dưới dạng bản đồ nhiệt, nhưng một số giá trị là NaN. Tôi muốn NaN được hiển thị dưới dạng màu đặc biệt không có trong bản đồ màu.

thí dụ:

import numpy as np
import matplotlib.pyplot as plt
f = plt.figure()
ax = f.add_subplot(111)
a = np.arange(25).reshape((5,5)).astype(float)
a[3,:] = np.nan
ax.imshow(a, interpolation='nearest')
f.canvas.draw()

Hình ảnh kết quả bất ngờ toàn màu xanh lam (màu thấp nhất trong biểu đồ màu máy bay phản lực). Tuy nhiên, nếu tôi thực hiện âm mưu như thế này:

ax.imshow(a, interpolation='nearest', vmin=0, vmax=24)

--Sau đó, tôi nhận được thứ gì đó tốt hơn, nhưng các giá trị NaN được vẽ cùng màu với vmin ... Có cách nào duyên dáng để tôi có thể đặt các NaN được vẽ bằng màu đặc biệt (ví dụ: xám hoặc trong suốt) không?

Với các phiên bản mới hơn của Matplotlib, không cần thiết phải sử dụng mảng che nữa.

Ví dụ: hãy tạo một mảng trong đó mọi giá trị thứ 7 là một NaN:

arr = np.arange(100, dtype=float).reshape(10, 10)
arr[~(arr % 7).astype(bool)] = np.nan

Chúng ta có thể sửa đổi bản đồ màu hiện tại và vẽ mảng bằng các dòng sau:

current_cmap = matplotlib.cm.get_cmap()
current_cmap.set_bad(color='red')
plt.imshow(arr)

Hrm, có vẻ như tôi có thể sử dụng một mảng được che để làm điều này:

masked_array = np.ma.array (a, mask=np.isnan(a))
cmap = matplotlib.cm.jet
cmap.set_bad('white',1.)
ax.imshow(masked_array, interpolation='nearest', cmap=cmap)

Điều này là đủ, mặc dù tôi vẫn mở các đề xuất. :]