2015-10-26 129 views
2

试图写一个脚本,将(最终)美国地图颜色红色或蓝色取决于州州长是共和党还是民主党。在尝试实际着色地图之前,我需要从dict'states'中得到给定状态的'r'或'd'值。基本上,我无法匹配从stateid = (p['id'])返回的文本字符串与字典states中同名键的值。任何人都可以帮我弄清楚如何做到这一点?遇到麻烦匹配字典键值

import BeautifulSoup 

states = {'AL':'r', 'AK':'d', 'AZ':'r', 'AR':'r', 'CA':'d', 'CO':'d', 'CT':'d','DE':'d','FL':'r','GA':'r','HI':'d','ID':'r','IL':'r','IN':'r','IO':'r','KA':'r','KY':'d','LA':'r','ME':'r','MD':'r','MA':'r','MI':'r','MN':'d','MS':'r','MO':'d','MT':'d','NE':'r','NH':'d','NJ':'r','NM':'r','NY':'d','NC':'r','ND':'r','OH':'r','OK':'r','OR':'d','PA':'d','RI':'d','SC':'r','SD':'r','TE':'r','TX':'r','UT':'r','VT':'d','VA':'d','WA':'d','WV':'d','WI':'r','WY':'r'} 

svg = open('blankmap.svg', 'r').read() 

soup = BeautifulSoup(svg, "lxml") 

paths = soup.find_all('path') 
for p in paths: 
    stateid = (p['id']) 
    if stateid in states: 
     print([stateid].values()) 

您可以在最后一行看到我的(失败)尝试完成此操作。

回答

4

你不只是意味着:

if stateid in states: 
    print(states[stateid]) 
+0

...哇。我现在觉得很愚蠢。感谢@alecxe! – n1c9