新建一个Excel文件(瞎编):population.xlsx,并进行读取等相关操作
'''
一些补充:
excel---->workbook---->worksheet---->(row,column,cell)
'''
import openpyxl
import pprint #打印数据的模块,美观打印数据
#读取出来的数据用字典数据结构存储
'''
数据结构核心:
{'中国':{'北京':{'score2':3,'socre1':189}, #把score相加
'上海':{....................}},
'美国':{ .......................}
'日本':{ .......................}
}
'''
print('打开读取excel文件')
wb=openpyxl.load_workbook('population.xlsx')
sheet=wb.active
data={} #输出的数据
'''
setdefault:字典类型的方法函数
picnic={'apple':5,'cup':10}
picnic.setdefault('orange',12) #picnic={'apple':5,'cup':10,'orange',12}
#如果有字典中本来有apple,则会忽略25这个缺省值,不会改变
picnic.setdefault('apple',25) #picnic={'apple':5,'cup':10,'orange',12}
'''
#遍历表单
for per_row in range(2,sheet.max_row+1):#第一行为头,没有数据,+1是因为range是左闭右开的
country=sheet['A'+str(per_row)].value #str(per_row)将其转换为字符串,进行字符串相加
city=sheet['B'+str(per_row)].value
score1=sheet['C'+str(per_row)].value
score2=sheet['D'+str(per_row)].value
x=score1
y=score2
data.setdefault(country,{}) #保证country的值存在
#下面的'score1' 'socre2'是字符串,与上面的score1 socre2不同
data[country].setdefault(city,{'score1':0,'socre2':0}) #{score1:0,socre2:0}与前一句的{}类似,还为两个值赋了初值0,方便后面的相加
data[country][city]['score1']+=int(score1) #累加过程,要使用int类型装换
data[country][city]['socre2']+=int(score2)
print(data)
'''
{'中国': {'北京': {'score1': 189, 'socre2': 3}, '上海': {'score1': 190, 'socre2': 7}},
'美国': {'WT.DC': {'score1': 183, 'socre2': 11}, 'NY': {'score1': 191, 'socre2': 15}},
'日本': {'TK': {'score1': 189, 'socre2': 19}, '大阪': {'score1': 178, 'socre2': 23}}}
'''
#将数据写入文本
print('Writing result.....')
resultfile=open('theresult.py','w') #保存到了一个.py的文件里面,也可以放到word(.doc)或excel(.xlsx)文件里面
'''
使用.py文件可以方便调用。
import resultfile
print(resultfile.allData['中国']['北京']['socre2'])
'''
resultfile.write('the result ='+ pprint.pformat(data)) #pprint.pformat(data):将其输出为字符串
'''
the result ={'中国': {'上海': {'score1': 190, 'socre2': 7}, '北京': {'score1': 189, 'socre2': 3}},
'日本': {'TK': {'score1': 189, 'socre2': 19},
'大阪': {'score1': 178, 'socre2': 23}},
'美国': {'NY': {'score1': 191, 'socre2': 15},
'WT.DC': {'score1': 183, 'socre2': 11}}}
'''
resultfile.close() #之前没有关闭这个语句,文件始终为空,排查后发现是没有关闭文件的语句