P129 csv文件计算

描述

在附件salary.in中存储了员工一季度1，2，3月的收入，编写程序计算一季度的总收入，并将结果按总收入降序写至salary.out文件。

格式

输入

salary.in文件，csv格式，utf-8编码

输出

salary.out文件，csv格式，utf-8编码

样例

salary.in文件内容

工号,1月,2月,3月
5101,18836,14436,7126
5102,19938,16542,11393
5103,19378,7600,10038
5104,5738,7646,10380
5105,19338,14081,8103
5106,11020,10326,6758
5107,18047,7297,10500
5108,9751,18908,8086
5109,12806,19817,5213
5110,15984,5373,5478

salary.out文件内容

工号,1月,2月,3月,一季度总收入
5102,19938,16542,11393,47873
5105,19338,14081,8103,41522
5101,18836,14436,7126,40398
5109,12806,19817,5213,37836
5103,19378,7600,10038,37016
5108,9751,18908,8086,36745
5107,18047,7297,10500,35844
5106,11020,10326,6758,28104
5110,15984,5373,5478,26835
5104,5738,7646,10380,23764

Accept代码

fr = open('salary.in', 'r', encoding='utf-8')
fw = open('salary.out', 'w', encoding='utf-8')
lst = []
for line in fr:
    line = line.replace('\n', '')
    lst.append(line.split(','))
fr.close()
lst[0].append("一季度总收入")
for i in range(1, len(lst)):
    total_income = 0
    for j in range(1, len(lst[i])):
        total_income += int(lst[i][j])
    lst[i].append(str(total_income))
lst[1:] = sorted(lst[1:], key=lambda x: int(x[-1]), reverse=True)
for item in lst:
    fw.write(','.join(item) + '\n')
fw.close()