Python作业五

作业五、文件操作
作业内容：
1、编写一个程序demo.py，要求运行该程序后，生成demo_new.py文件，其中内容与demo.py一样，只是在每一行的后面加上行号。要求行号以#开始，并且所有行的#垂直对齐。
2、假设某学校所有课程每学期允许多次考试，学生可随时参加考试，系统自动将每次考试的成绩都添加到Excel文件（包含3列：姓名、课程、成绩）中，现期末要求统计所有学生每门课程的最高成绩 。
编写程序，模拟生成若干同学的成绩 并写入Excel文件，其中学生姓名和课程名称均可重复，也就是允许出现同一门课程的多次成绩，最后统计所有学生每门课程的最高成绩，并写入新的Excel文件。

#5.1
filename = ('C:/Users/Lenovo/Desktop/demo.py');#避免与转义字符混淆，这里用除号
with open(filename,'r') as fp:#以只读的方式打开该文件，with:自动关闭,文件中有汉字会导致编码出现问题
    list1 = fp.readlines();#从文件读取整行，可作列表使用，使用前提是open模式为'r'
mxlen = len(max(list1,key=len));#长度为规则
list1 = [line.rstrip().ljust(mxlen)+'#'+str(index)+'\n'for index, line in enumerate(list1)];
with open(filename[:-3]+'_new.py','w') as fp:#创建文件的格式：open(‘path’+filename+‘文件格式’,‘w’)
    fp.writelines(list1);#[:-3] 切片，截取filename文件名的[0~-3]
print(filename);

#5.2
import openpyxl ##对excel操作
import random
def autodata(fileName): #生成数据
    wb = openpyxl.Workbook();
    sheet = wb.active;
    sheet.append(['姓名','课程','成绩'])
    first = tuple('肖杨高唐谭沈');
    middle = tuple('二三四五六七八');
    subjects = ('c++','data structure','python');
    for i in range(500):
        line = [];#一行
        name = random.choice(first)+random.choice(middle);
        line.append(name);
        line.append(random.choice(subjects));
        line.append(random.randint(60,100));
        sheet.append(line);
    wb.save(fileName);

def getMaxScore(oldfile,newfile):
    wb = openpyxl.load_workbook(oldfile);
    final_wb = openpyxl.Workbook();#大写创建excel
    start_sheet = wb.active;#通过active获取正在将使用的工作表
    final_sheet = final_wb.active;
    final_sheet.title = '学生各科成绩';
    final_sheet.append(['姓名','课程','最高分']);
    stuGrade = {}#创建记录全校学生成绩的字典
    for i in range(2,start_sheet.max_row+1):
        stuName=start_sheet['A'+str(i)].value #获取学生名字,下同
        lesName=start_sheet['B'+str(i)].value #str()将对象转换成字符串
        score=start_sheet['C'+str(i)].value  #获取excel中的信息：sheet[列+行]列为大写英文字母，行为数字，但都是字符串
        if stuName in stuGrade:
            stuGrade[stuName]#do nothing
        else:
            stuGrade[stuName] = {};
        if lesName in stuGrade[stuName]:
            if score>stuGrade[stuName][lesName]:
                 stuGrade[stuName][lesName]=score;
        else:
            stuGrade[stuName][lesName]=score;
    #将最总成绩汇入新表，按姓名有序，如果在此基础上课程有序应再加一循环访问课程
    for student in stuGrade:
        for lesson,score in stuGrade[student].items():
            final_sheet.append([student,lesson,score]);
    final_wb.save(newfile);
oldfile = '成绩.xlsx';
newfile = '最终成绩.xlsx';
autodata(oldfile);
getMaxScore(oldfile, newfile);