Matplotlib的“ ax1.add_collection(mylinecollection)”的底图等效项是什么?

时间:2019-03-26 21:39:52

标签: python python-2.7 matplotlib matplotlib-basemap

我正在从关系数据库中提取纬度/经度。这些点是线段的顶点,并且组织了数据库表,以便可以根据线号对点进行分组,并根据点序号进行排序。具体来说,数据库表具有以下字段:纬度,经度,line_number,pt_order_number。

我已将所有线段拉入线集合。使用Matplotlib ax1.add_collection(mylinecollection)可以按预期进行绘制。请注意,我使用线集合而不是ax1.plot(long,lat)循环,因为某些数据集具有数千条线和数万个点(线集合更快,更干净)。

但是,我需要使用Basemap绘制这些线集合(这样我才能获得漂亮的阴影浮雕和比例尺)。底图无法通过.add_collection(),因此我不确定替代方法是什么。我在底图教程中找不到答案或明显的替代方法(但也许我错过了)。感谢您的帮助。

当前代码:

# note linecollection variable produced in function; performs as expected with "ax1.add_collection(linecollection)"
latitude_range = [34.012965603200001, 34.721878621999998]
longitude_range = [-116.76151759999999, -116.33691841200002]

import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection
from mpl_toolkits.basemap import Basemap

south, north = latitude_range[0], latitude_range[1]
west, east = longitude_range[0], longitude_range[1]
center = [(east+west)/2, (north+south)/2]

m = Basemap(llcrnrlon=west, llcrnrlat=south, urcrnrlon=east, urcrnrlat=north,
            resolution='c', epsg=4326,
            lon_0=center[0], lat_0=center[1])

m.arcgisimage(service = "World_Shaded_Relief", xpixels = 2000)
m.add_collection(linecollection) # throws error "AttributeError: 'Basemap' object has no attribute 'add_collection'"

plt.show()

1 个答案:

答案 0 :(得分:1)

提供以上评论作为答案:

底图使用轴:

fig, ax = plt.subplots()
m = Basemap(..., ax=ax) 

在此轴上,您可以调用任何喜欢的方法,例如

ax.add_collection(...)

确保首先将线集合的坐标转换为底图单位。