如何使用Python在Basemap中添加图例
我想在我的世界地图上画一个传奇。每个点都是来自不同tweet的不同类型的设备。 这是我的密码:如何使用Python在Basemap中添加图例,python,python-2.7,matplotlib,matplotlib-basemap,Python,Python 2.7,Matplotlib,Matplotlib Basemap,我想在我的世界地图上画一个传奇。每个点都是来自不同tweet的不同类型的设备。 这是我的密码: import geocoder import json from mpl_toolkits.basemap import Basemap import matplotlib.pyplot as plt def openJSON(file, path): with open(path + file + '.json', 'r') as fp: data = json.load
import geocoder
import json
from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt
def openJSON(file, path):
with open(path + file + '.json', 'r') as fp:
data = json.load(fp)
return data
def App_users(data):
android_users = []
iphone_users = []
other_users = []
#3 types of users: Android, Iphone and others:
for a, b, c, d in data:
if "Twitter for Android" in d:
android_users.append([a, b])
elif "Twitter for iPhone" in d:
iphone_users.append([a, b])
else:
other_users.append([a, b])
data = (android_users, iphone_users, other_users)
return data
if __name__ == '__main__':
posXY_add = []
for d in data[0:50]:
if d["user"]["location"] != None:
g = geocoder.google(d["user"]["location"])
if g.latlng != None:
a = g.latlng + [g.address] + [str(d["source"])]
posXY_add.append(a)
data = App_users(posXY_add)
map = Basemap(llcrnrlat=-70, urcrnrlat=70,
llcrnrlon=-180, urcrnrlon=180, epsg=3395)
map.arcgisimage(service='ESRI_Imagery_World_2D', xpixels = 1500, verbose= True)
colors = ['r', 'g', 'b']
increment = 0
for app in data:
for item in app:
if len(item) == 2:
lat, lon = item[0], item[1]
x, y = map(lon, lat)
map.plot(x, y, colors[increment] + 'o')
increment += 1
plt.title("Geo-position Twitter Devices")
plt.show()
你有什么建议来改进我的地图吗?不同的颜色,不同的地图还是别的?
我正在使用Pycharm的Python2.7。我的样本数据是:
([[21.8974003, 83.39496319999999], [40.2671941, -86.1349019], [10.4805937, -66.90360629999999]], [[50.719164, -1.880769], [37.0903748, -94.5134078], [44.9862701, -93.39398279999999], [47.6062095, -122.3320708], [-27.4697707, 153.0251235], [20.593684, 78.96288]], [[43.2648545, -79.9492978], [39.1767262, -94.4867155], [33.7489954, -84.3879824], [39.7392358, -104.990251], [38.8761634, -77.0337777], [41.0082376, 28.9783589], [36.0753128, -95.95646269999999], [36.0595387, -95.90582429999999], [29.7604267, -95.3698028], [21.3014947, 106.6291304], [21.3014947, 106.6291304], [-16.6868982, -49.2648114], [10.4805937, -66.90360629999999], [45.037149, -92.825929], [33.760818, -78.967556], [45.49340369999999, -73.82329179999999], [39.750545, 37.0150217]])
图例部分:这与将图例放入绘图的方式相同-只需使用
m.plot()
调用label
参数即可。因此,在您的情况下,它将是:
handle_dict = {
"Andriod" : 0,
"iPhone" : 0,
"Other" : 0
}
colors = ['r', 'g', 'b']
increment = 0
for app in data:
for item in app:
if len(item) == 2:
lat, lon = item[0], item[1]
x, y = map(lon, lat)
map.plot(x, y, colors[increment] + 'o', label =
handle_dict[type_phone] if handle_dict[type_phone] == 0
else "_no-legend_")
handle_dict[type_phone] += 1
increment += 1
plt.legend()
plt.title("Geo-position Twitter Devices")
plt.show()
您还必须修改代码以识别正在打印的电话类型(因此变量type\u phone
)
以下是有关图例的链接,用于修改图例
改善地图:
就我个人而言,我喜欢使用map.shadeDelief()
来查看这些世界视图-它不像map.bluemarble()
那样暗,而且你可以很容易地更好地看到这些点。这里真的不需要map.arcgisimage()
参数,因为您有一个世界视图而不是放大视图-如果使用内置函数,它将更快地生成地图。我还将markeredgewidth
参数添加到map.plot()
中,以使标记边宽度不是零,因为现在点看起来像是与背景混合在一起。它会让你的圆点更加突出
哦-如果可以的话,也可以花点时间升级到Python 3。我刚刚用一个技巧解决了这个问题。在“a”上对我的代码进行迭代;通过这种方式,我只绘制类别的第一个标签。代码:
handles = [("Android",None), ("iPhone",None), ("Other devices",None)]
colors = ['r', 'g','b']
increment = 0
for app in data:
a = 0
for item in app:
if len(item) == 2:
lat, lon = item[0], item[1]
x, y = map(lon, lat)
map.plot(x, y, colors[increment] + 'o', alpha=0.8, markersize=0.1, label=handles[increment][a])
a = 1
increment += 1
plt.legend(loc='lower left', shadow=True)
谢谢你的回复,我很感激。不幸的是,这个问题还没有得到解决,因为我在图例中多次看到标签。所以我的传奇中有很多带有“Android”的红点。我不能只展示三个标签。啊——我这边的疏忽。我继续编辑了这篇文章,在
map.plot()
中添加了label=handle\u dict[type\u phone],如果handle\u dict[type\u phone]==0,则在map.plot()
中添加了“\u no-legend”
,并添加到了词典中。非常感谢,这太棒了。