由一道python的面试题说起

关于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 collectionsmat=[[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 DeDictprint sorted(DeDict.items(),cmp=None,key=lambda x:x[1][0])