sorted, itemgetter, defaultdict

Posted 2020-09-10

defaultdict用法：

import collections
s = [(‘yellow‘, 1), (‘blue‘, 2), (‘yellow‘, 3), (‘blue‘, 4), (‘red‘, 1)]
# defaultdict
d = collections.defaultdict(list)
for k, v in s:
    d[k].append(v)
# Use dict and setdefault   
g = {}
for k, v in s:
    g.setdefault(k, []).append(v)
      
# Use dict
e = {}
for k, v in s:
    e[k] = v

 结果：

list(d.items())
[(‘blue‘, [2, 4]), (‘red‘, [1]), (‘yellow‘, [1, 3])]
>>> list(g.items())
[(‘blue‘, [2, 4]), (‘red‘, [1]), (‘yellow‘, [1, 3])]
>>> list(e.items())
[(‘blue‘, 4), (‘red‘, 1), (‘yellow‘, 3)]
>>> d
defaultdict(<class ‘list‘>, {‘blue‘: [2, 4], ‘red‘: [1], ‘yellow‘: [1, 3]})
>>> g
{‘blue‘: [2, 4], ‘red‘: [1], ‘yellow‘: [1, 3]}
>>> e
{‘blue‘: 4, ‘red‘: 1, ‘yellow‘: 3}
>>> d.items()
dict_items([(‘blue‘, [2, 4]), (‘red‘, [1]), (‘yellow‘, [1, 3])])
>>> d["blue"]
[2, 4]
>>> d.keys()
dict_keys([‘blue‘, ‘red‘, ‘yellow‘])
>>> d.default_factory
<class ‘list‘>
>>> d.values()
dict_values([[2, 4], [1], [1, 3]])

 这里的defaultdict(function_factory)构建的是一个类似dictionary的对象，其中keys的值，自行确定赋值，但是values的类型，是function_factory的类实例，而且具有默认值。比如default(int)则创建一个类似dictionary对象，里面任何的values都是int的实例，而且就算是一个不存在的key, d[key] 也有一个默认值，这个默认值是int()的默认值0.

sorted用法：

--------------------------------sorted---------------------------------------
>>> help(sorted)
Help on built-in function sorted in module __builtin__:

sorted(...)
    sorted(iterable, cmp=None, key=None, reverse=False) --> new sorted list
---------------------------------sort----------------------------------------
>>> help(list.sort)
Help on method_descriptor:

sort(...)
    L.sort(cmp=None, key=None, reverse=False) -- stable sort *IN PLACE*;
    cmp(x, y) -> -1, 0, 1
-----------------------------------------------------------------------------

 

参数解释：

 

（1）iterable指定要排序的list或者iterable，不用多说；

 

（2）cmp为函数，指定排序时进行比较的函数，可以指定一个函数或者lambda函数，如：

      students为类对象的list，没个成员有三个域，用sorted进行比较时可以自己定cmp函数，例如这里要通过比较第三个数据成员来排序，代码可以这样写：
      students = [(‘john‘, ‘A‘, 15), (‘jane‘, ‘B‘, 12), (‘dave‘, ‘B‘, 10)]
      sorted(students, key=lambda student : student[2])

（3）key为函数，指定取待排序元素的哪一项进行排序，函数用上面的例子来说明，代码如下：
　　sorted(students, key=lambda student : student[2])

key指定的lambda函数功能是去元素student的第三个域（即：student[2]），因此sorted排序时，会以students所有元素的第三个域来进行排序。

有了上面的operator.itemgetter函数，也可以用该函数来实现，例如要通过student的第三个域排序，可以这么写：
sorted(students, key=operator.itemgetter(2)) 
sorted函数也可以进行多级排序，例如要根据第二个域和第三个域进行排序，可以这么写：
sorted(students, key=operator.itemgetter(1,2))

 

即先跟句第二个域排序，再根据第三个域排序。
（4）reverse参数就不用多说了，是一个bool变量，表示升序还是降序排列，默认为false（升序排列），定义为True时将按降序排列。

 

 

sorted产生一个新值，sort在原列表上进行操作

 

itemgetter用法

字典的items()函数返回包含字典所有元素的列表。

operator模块提供的itemgetter函数用于获取对象的哪些维的数据，参数为一些序号（即需要获取的数据在对象中的序号）

要注意，operator.itemgetter函数获取的不是值，而是定义了一个函数，通过该函数作用到对象上才能获取值。