Python 用 openpyxl 模块统计 Excel 表格中的数据，以字典形式写入 py 文件

1、加载模块，设置路径，读取表格

import os
import openpyxl
import pprint

os.chdir('D:\\pySpace')
print('Opening workbook...')
wb = openpyxl.load_workbook('censuspopdata.xlsx')    # 加载电子表格
sheet = wb.get_sheet_by_name('Population by Census Tract')
countyData = {}

 

 2、f遍历所有的行， 汇总各州的每个县的统计地区数和每个县的人口数，将数据填充到字典 countyData 中。

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)

 

 3、创建 py 文件，并将之前生成的字典数据写入其中。

print('Wring results...')
resultFile = open('census2010.csv', 'w')
resultFile.write('allData = ' + pprint.pformat(countyData))
resultFile.close()
print('Done.')