python绘制格网图显示格网，并把nan值设为黑色

你可以使用 matplotlib 库绘制格网图，并使用 numpy 中的 `numpy.ma.masked_invalid()` 方法将 NaN 值设置为黑色。示例代码如下：

import numpy as np
import matplotlib.pyplot as plt

# 创建示例数据
data = np.random.rand(10, 10)
data[3, 5] = np.nan

# 绘制格网图
fig, ax = plt.subplots()
im = ax.imshow(data, cmap='gray')

# 显示格网
ax.set_xticks(np.arange(-.5, 10, 1))
ax.set_yticks(np.arange(-.5, 10, 1))
ax.grid(color='white', linewidth=2)

# 将 NaN 值设置为黑色
data = np.ma.masked_invalid(data)
im.set_array(data)

# 显示图像
plt.show()

以上代码中，`np.ma.masked_invalid()` 方法可以将 NaN 值设置为遮罩值，然后通过 `imshow()` 方法将遮罩值设置为黑色。使用 `grid()` 方法可以显示格网线，其中 `color` 参数指定线的颜色，`linewidth` 参数指定线的宽度。