matplotlib - 如何在底图上绘制矩形

Question

我正在寻找一种在底图上绘制填充矩形的方法。我可以使用该方法轻松绘制矩形的边缘drawgreatcircle，但我找不到实际填充这些矩形的方法（指定颜色和 alpha）。

Answer 1

您可以将 matplotlib.patches.Polygon() 直接添加到您的轴上。问题是您是否希望矩形定义绘图坐标（绘图上的直线）或地图坐标（绘图上的大圆圈）。无论哪种方式，您都可以在地图坐标中指定顶点，然后通过调用 Basemap 实例（m()在下面的示例中）将它们转换为绘图坐标，自己构建一个多边形，然后手动将其添加到要渲染的轴上。

对于在绘图坐标中定义的矩形，这是一个示例：

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

def draw_screen_poly( lats, lons, m):
    x, y = m( lons, lats )
    xy = zip(x,y)
    poly = Polygon( xy, facecolor='red', alpha=0.4 )
    plt.gca().add_patch(poly)

lats = [ -30, 30, 30, -30 ]
lons = [ -50, -50, 50, 50 ]

m = Basemap(projection='sinu',lon_0=0)
m.drawcoastlines()
m.drawmapboundary()
draw_screen_poly( lats, lons, m )

plt.show()

对于在地图坐标中定义的矩形，使用相同的方法，但在转换为绘图坐标之前在地图空间中插入线。对于每个线段，您必须执行以下操作：

lats = np.linspace( lat0, lat1, resolution )
lons = np.linspace( lon0, lon1, resolution )

然后将这些地图坐标转换为绘图坐标（如上，使用m()）并再次使用绘图坐标创建一个多边形。

Answer 2

使用安德鲁的答案，我得到了错误

TypeError: len() of unsized object.

但是，将 zip 转换为列表可以解决此问题。

完整代码：

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

def draw_screen_poly( lats, lons, m):
    x, y = m( lons, lats )
    xy = zip(x,y)
    poly = Polygon( list(xy), facecolor='red', alpha=0.4 )
    plt.gca().add_patch(poly)

lats = [ -30, 30, 30, -30 ]
lons = [ -50, -50, 50, 50 ]

m = Basemap(projection='sinu',lon_0=0)
m.drawcoastlines()
m.drawmapboundary()
draw_screen_poly( lats, lons, m )

plt.show()

Answer 3

与上面类似的答案，但更基本的代码：

from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt
from matplotlib.patches import Polygon

map = Basemap(projection='cyl')

map.drawmapboundary(fill_color='aqua')
map.fillcontinents(color='coral',lake_color='aqua')
map.drawcoastlines()

x1,y1 = map(-25,-25)
x2,y2 = map(-25,25)
x3,y3 = map(25,25)
x4,y4 = map(25,-25)
poly = Polygon([(x1,y1),(x2,y2),(x3,y3),(x4,y4)],facecolor='red',edgecolor='green',linewidth=3)
plt.gca().add_patch(poly)

plt.show()