根据带有Basemap的字典值来遮盖国家/地区的状态

时间:2016-03-20 20:17:32

标签: python shapefile matplotlib-basemap

我想绘制墨西哥的地图,并根据字典的值对着状态进行着色。我使用了上一个问题(Easiest way to plot data on country map with python)中建议的以下代码,到目前为止它绘制了国家和州,但是当我尝试定义着色时,我得到一个错误。代码下方:

import numpy as np
    import matplotlib.pyplot as plt
    from matplotlib.collections import PatchCollection
    from mpl_toolkits.basemap import Basemap
    %matplotlib inline
    from shapely.geometry import Polygon

mexican_states_people = {'AGS': 20,'BC': 57, 'BCS': 562, 'CAMP': 594,'CHIH': 442,'CHIS': 69,'COAH': 100,'COL': 237,'DF': 7323,'DGO': 689,'GRO': 40,'GTO': 295,'HGO': 1134,'JAL': 875,'MEX': 1,'MICH': 393, 'MOR': 301,'NAY': 404,'NL': 327,'OAX': 391,'PUE': 670,'QRO': 270,'QROO': 156,'SIN': 63,'SLP': 689,'SON': 291,'TAB': 306,'TAMPS': 59,'TLAX': 108,'VER': 17,'YUC': 35,'ZAC': 890}

m = Basemap(llcrnrlon=-115,llcrnrlat=5,urcrnrlon=-80,urcrnrlat=35,
            resolution='i',projection='tmerc',lon_0=-99,lat_0=19)
m.readshapefile("MEX_adm1", "mexican_states")

max_people = np.max(mexican_states_people.values())

for coordinates, state in zip(m.mexican_states, m.mexican_states_info):
    print state
    if state["State_name"] in mexican_states_people.keys():
        shade = mexican_states_people[state["State_name"]]/max_people


m.drawcoastlines()
m.fillcontinents(color='coral',lake_color='aqua')
m.drawparallels(np.arange(-40,61.,2.))
m.drawmeridians(np.arange(-20.,21.,2.))
m.drawmapboundary(fill_color='aqua')
plt.title("Mexico")
plt.show()

我收到错误:KeyError: 'State_name'print state命令提供了以下信息:

{'NAME_0': 'Mexico', 'NAME_1': 'Aguascalientes', 'TYPE_1': 'Estado', 'CCA_1': '                                                                                                                                                                                                                                                              ', 'VARNAME_1': '                                                                                                                                                      ', 'ENGTYPE_1': 'State', 'HASC_1': 'MX.AG', 'RINGNUM': 1, 'ID_0': 145, 'ID_1': 1, 'ISO': 'MEX', 'NL_NAME_1': '                                                  ', 'CCN_1': 0, 'SHAPENUM': 1}

http://www.gadm.org/

下载了shapefile“MEX_adm1”

1 个答案:

答案 0 :(得分:2)

您从我之前的回答中直接复制了:)

如果您将if re.match("c", "cat"): print "Yes!" 替换为if state["State_name"],这应该有效,因为shapefile中的名称的关键字是if state["NAME_1"] - 但是会有一个不同的问题,因为显然是{{1} dict使用与shapefile不同的命名约定,在你的dict中,state缩写为“AGS”,而在shapefile中,它被称为“Aguascalientes”。

如果状态不是太多,你可以手动更改你的字典,否则你必须在缩写和名字之间找到一些其他的映射。