自动化无聊的东西excel，不能得到代码来拉动census2010 ['AK']

Question

我正在遵循第13章下的示例项目(从电子表格中读取数据)。我已经创建了人口普查2010.py代码并将其保存在我的文档中。当我运行引号时，我会得到这个键错误。

KeyError                                  Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_35244/1301298642.py in <module>
     26 
     27 import census2010
---> 28 census2010.allData['AK']['Anchorage']
     29 anchoragePop = census2010.allData['AK']['Anchorage']['pop']
     30 print('The 2010 population of Anchorage was ' + str(anchoragePop))

KeyError: 'AK'

这是我在交互式shell上运行的当前代码。

import openpyxl, pprint
import os
os.chdir('C:\\users\\brian\\documents')
print('Opening workbook...')

wb = openpyxl.load_workbook('censuspopdata.xlsx')
sheet = wb['Population by Census Tract']
countyData = {}

# TODO: Fill in countyData with each county's population and tracts.
print('Reading rows...')

for row in range(2, sheet.max_row + 1):
    state = sheet['B' + str(row)].value
    county = sheet['C' + str(row)].value
    pop = sheet['D' + str(row)].value
    countyData.setdefault(state, {})
    countyData[state].setdefault(county, {'tracts': 0, 'pop': 0})
    countyData[state][county]['tracts'] += 1
    countyData[state][county]['pop'] += int(pop)
print('Writing results...')
resultFile = open('census2010.py', 'w')
resultFile.write('allData = ' + pprint.pformat(countyData))
resultFile.close()
print('Done.')

import census2010
census2010.allData['AK']['Anchorage']
anchoragePop = census2010.allData['AK']['Anchorage']['pop']
print('The 2010 population of Anchorage was ' + str(anchoragePop))

不确定如何从也保存在文档中的excel文件中获取数据。

Answer 1

是否可能将当前目录更改为

os.chdir('C:\users\brian\documents')

要下载excel数据，在这种情况下，脚本将编写一个名为普查2010.py的文件到该路径。但是，在您使用交互式shell的同一个目录中，也有一个名为普查2010.py的文件。在这种情况下，您需要使用

import sys
sys.path.insert(1, 'C:\\users\\brian\\documents')

import census2010

而不是仅仅

import census2010

Importing files from different folder