Set mask for matplotlib tricontourf

后端 未结 2 1103
轻奢々
轻奢々 2021-01-19 19:27

I have some numpy array containing data that I would visualize on a 2D grid. Some of the data is unphysical and I would like to mask this data. However, I could not figure o

2条回答
  •  一生所求
    2021-01-19 19:51

    wsj's answer didn't work for me since it didn't remove certain masked points (I think when not all of the nodes where bad).

    This solution did:

    z[isbad] = numpy.NaN
    z = numpy.ma.masked_invalid(z)
    vmin, vmax = z.min(), z.max()
    z = z.filled(fill_value=-999)
    
    levels = numpy.linspace(vmin, vmax, n_points)
    plt.tricontourf(x, y, z, levels=levels)
    

提交回复
热议问题