pcolormesh 缺失值? [英] pcolormesh with missing values?

问题描述

我有 3 个一维数组:x, y, z

I have 3 1-D ndarrays: x, y, z

以及以下代码:

import numpy as np
import matplotlib.pyplot as plt
import scipy.interpolate as spinterp

## define data
npoints = 50
xreg = np.linspace(x.min(),x.max(),npoints)
yreg = np.linspace(y.min(),y.max(),npoints)
X,Y = np.meshgrid(xreg,yreg)
Z = spinterp.griddata(np.vstack((x,y)).T,z,(X,Y),
                      method='linear').reshape(X.shape)

## plot
plt.close()
ax = plt.axes()
col = ax.pcolormesh(X,Y,Z.T)
plt.draw()

我的情节是空白的，我怀疑这是因为 method='linear' 插值与 nans 一起出现.我试过转换为掩码数组，但无济于事 - 情节仍然是空白的.你能告诉我我做错了什么吗?谢谢.

My plot comes out blank, and I suspect it is because the method='linear' interpolation comes out with nans. I've tried converting to a masked array, but to no avail - plot is still blank. Can you tell me what I am doing wrong? Thanks.

推荐答案

明白了.这似乎有点绕，但这就是解决方案:

Got it. This seems round-about, but this was the solution:

import numpy.ma as ma

Zm = ma.masked_where(np.isnan(Z),Z)
plt.pcolormesh(X,Y,Zm.T)

如果Z矩阵包含nan的，它必须是pcolormesh的掩码数组，必须用ma.masked_where创建代码>，或者，

If the Z matrix contains nan's, it has to be a masked array for pcolormesh, which has to be created with ma.masked_where, or, alternatively,

Zm = ma.array(Z,mask=np.isnan(Z))

这篇关于pcolormesh 缺失值?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！