生成excel文件和随机数据，遍历数据，对数据按规则处理后生成新的excel

admin

生成excel文件和随机数据，遍历数据，对数据按规则处理后生成新的excel。代码：

1. import openpyxl
   
2. from openpyxl import Workbook
   
3. import random
   
4. 
   
5. #生成随机数据
   
6. def generateRandomInformation(filename):
   
7.     workbook = Workbook()
   
8.     worksheet = workbook.worksheets[0]
   
9.     worksheet.append(['姓名','课程','成绩'])
   
10. 
    
11.     #中文名字中的第一、第二、第三个字
    
12.     first = '赵钱孙李'
    
13.     middle = '伟昀琛东'
    
14.     last = '坤艳志'
    
15.     subjects = ('语文','数学','英语')
    
16.     for i in range(200):
    
17.         line = []
    
18.         r = random.randint(1,100)
    
19.         name = random.choice(first)
    
20.         #按一定概率生成只有两个字的中文名字
    
21.         if r>50:
    
22.             name = name + random.choice(middle)
    
23.         name = name + random.choice(last)
    
24.         #依次生成姓名、课程名称和成绩
    
25.         line.append(name)
    
26.         line.append(random.choice(subjects))
    
27.         line.append(random.randint(0,100))
    
28. 
    
29.         worksheet.append(line)
    
30.     #保存数据，生成Excel 2007格式的文件
    
31.     workbook.save(filename)
    
32. 
    
33. def getResult(oldfile, newfile):
    
34.     #用于存放结果数据的字典
    
35.     result = dict()
    
36. 
    
37.     #打开原始数据
    
38.     workbook = openpyxl.load_workbook(oldfile)
    
39.     worksheet = workbook.worksheets[0]
    
40. 
    
41.     #遍历原始数据
    
42.     for row in worksheet.rows:
    
43.         if row[0].value == '姓名':
    
44.             continue
    
45.         #姓名,课程名称,本次成绩
    
46.         name, subject, grade = row[0].value, row[1].value, row[2].value
    
47. 
    
48.         #获取当前姓名对应的课程名称和成绩信息
    
49.         #如果result字典中不包含，则返回空字典
    
50.         t = result.get(name, {})
    
51.         #获取当前学生当前课程的成绩，若不存在，返回0
    
52.         f = t.get(subject, 0)
    
53.         #只保留该学生该课程的最高成绩
    
54.         if grade > f:
    
55.             t[subject] = grade
    
56.             result[name] = t
    
57. 
    
58.     workbook1 = Workbook()
    
59.     worksheet1 = workbook1.worksheets[0]
    
60.     worksheet1.append(['姓名','课程','成绩'])
    
61. 
    
62.     #将result字典中的结果数据写入Excel文件
    
63.     for name, t in result.items():
    
64.         print(name,t)
    
65.         for subject, grade in t.items():
    
66.             worksheet1.append([name, subject, grade])
    
67. 
    
68.     workbook1.save(newfile)
    
69. 
    
70. if __name__ == '__main__':
    
71.     oldfile = r'test.xlsx'
    
72.     newfile = r'result.xlsx'
    
73.     generateRandomInformation(oldfile)
    
74.     getResult(oldfile, newfile)

复制代码

执行结果：生成一个老excel和一个新的excel。