关于sorted和defaultdict的使用
mat=[[1,38],[3,45],[4,31],[7,44],[2,76]]
chin=[[2,34],[1,78],[3,46],[7,88],[9,75],[11,32]]
题目大意:
mat[i][0]和chin[i][0]代表的是学生的学号,mat[i][1]和chin[i][1]分别存储的是学生的数学和语文的成绩,要求将这两组数据组合成一组数据(学号,数学成绩,语文成绩)然后按照要求进行排序(可能是按学号,也可能是按某一门的成绩)
考察点:
- 如何选择合适的数据结构,能够使得数据合并操作时候复杂度尽量低
- 如何对合并后的数据按照任意给定规则排序(这个显然是对sorted函数的考察)
选择list作为合并后的数据结构显然是不合适的,因为math和chin存在学号相同的情况,去重的工作量会比较大(因为实际这两个list可能会很大)
defaultdict显然是一个不错的选择,但如何使得defaultdict的缺省value是[0,0]呢,思考后,我采用了下面的写法:
import collections
mat=[[1,38],[3,45],[4,31],[7,44],[2,76]]
chin=[[2,34],[1,78],[3,46],[7,88],[9,75],[11,32]]
DeDict = collections.defaultdict(lambda:[0,0])
for i in mat:
DeDict[i[0]][0]=i[1]
for i in chin:
DeDict[i[0]][1]=i[1]
print DeDict
print sorted(DeDict.items(),cmp=None,key=lambda x:x[1][0])