python - python - 如何使用Basemap lib在Python中的contourf上绘制点？

Question

我正在寻找一种简单的方法来在轮廓底图图中的某些值上绘制点

这是一个示例代码：

from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt
import numpy as np

dx, dy = 0.5, 0.5

# generate 2 2d grids for the x & y bounds
y, x = np.mgrid[slice(-15, 15 + dy, dy),
                slice(-90, -60 + dx, dx)]

z = np.sin(x) + np.cos(5 + y*x) * np.cos(x) + x*y/100
n, m = z.shape
z = z - np.exp(np.random.rand(n,m))

#Setup the map
m = Basemap(projection='merc', llcrnrlat=-15, urcrnrlat=15,\
            llcrnrlon=-90, urcrnrlon=-60, resolution='l')
m.drawcoastlines()

xx, yy = m(x,y)
cs = m.contourf(xx,yy,z,cmap=plt.cm.Spectral_r)
cbar = plt.colorbar(cs, orientation='horizontal', shrink=0.5)

plt.show()

我想要a = np.where(abs(z) > 5,np.nan,z)Basemap 像附图中那样绘制点的地方

标签： pythonmatplotlib-basemap

Answer 1

您可以在第二次调用中使用舱口盖contourf()：

from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt
import numpy as np

dx, dy = 0.5, 0.5

# generate 2 2d grids for the x & y bounds
y, x = np.mgrid[slice(-15, 15 + dy, dy),
                slice(-90, -60 + dx, dx)]

z = np.sin(x) + np.cos(5 + y*x) * np.cos(x) + x*y/100
n, m = z.shape
z = z - np.exp(np.random.rand(n,m))

#Setup the map
m = Basemap(projection='merc', llcrnrlat=-15, urcrnrlat=15,\
            llcrnrlon=-90, urcrnrlon=-60, resolution='l')
m.drawcoastlines()

xx, yy = m(x,y)
cs = m.contourf(xx,yy,z,cmap=plt.cm.Spectral_r)
cbar = plt.colorbar(cs, orientation='horizontal', shrink=0.5)

##adding hatches:
cs = m.contourf(xx, yy, z, levels=[np.min(z),5,np.max(z)], colors='none',
                  hatches=[None,'.',],
                  extend='lower')

plt.show()

给出以下图像：