Python: 字典列表: itemgetter 函数: 根据某个或某几个字典字段来排序列表

时间:2023-12-17 08:51:14

问题:根据某个或某几个字典字段来排序Python列表

answer: 通过使用operator 模块的itemgetter 函数,可以非常容易的排序这样的数据结构

eg:

  rows = [
  {'fname': 'Brian', 'lname': 'Jones', 'uid': 1003},
  {'fname': 'David', 'lname': 'Beazley', 'uid': 1002},
  {'fname': 'John', 'lname': 'Cleese', 'uid': 1001},
  {'fname': 'Big', 'lname': 'Jones', 'uid': 1004}
  ]

  1.itemgetter()

  from operator import itemgetter

  1).print(rows_by_fname)

  >>>rows_by_fname = sorted(rows, key=itemgetter('fname'))

  >>>print(rows_by_fname)

  [{'fname': 'Big', 'uid': 1004, 'lname': 'Jones'},
  {'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'},
  {'fname': 'David', 'uid': 1002, 'lname': 'Beazley'},
  {'fname': 'John', 'uid': 1001, 'lname': 'Cleese'}]

  2).print(rows_by_uid) 

  >>>rows_by_uid = sorted(rows, key=itemgetter('uid'))

  >>>print(rows_by_uid)
  [{'fname': 'John', 'uid': 1001, 'lname': 'Cleese'},
  {'fname': 'David', 'uid': 1002, 'lname': 'Beazley'},
  {'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'},
  {'fname': 'Big', 'uid': 1004, 'lname': 'Jones'}]

  3).itemgetter() 函数也支持多个keys

  itemgetter() 函数也支持多个keys,比如下面的代码
  >>>rows_by_lfname = sorted(rows, key=itemgetter('lname','fname'))
  >>>print(rows_by_lfname)

  [{'fname': 'David', 'uid': 1002, 'lname': 'Beazley'},
  {'fname': 'John', 'uid': 1001, 'lname': 'Cleese'},
  {'fname': 'Big', 'uid': 1004, 'lname': 'Jones'},
  {'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'}]

  在上面例子中,rows被传递给接受一个关键字参数的sorted()内置函数。

  这个参数是callable类型,并且从rows中接受一个单一元素,然后返回被用来排序的值。

  itemgetter()就是负责创建这个callable对象的。

  如果传多个参数给itemgetter(),它生成的callabe对象会返回一个包含所有值的元组,sort()函数会根据元组中元素顺序去排序。

  2.lambda表达式

    itemgetter()有时候可以用lambda表达式代替

    rows_by_fname = sorted(rows, key=lambda r: r['fname'])
    rows_by_lfname = sorted(rows, key=lambda r: (r['lname'],r['fname']))

    这种方案也不错。但是,使用itemgetter() 方式会运行的稍微快点。因此,如果
    你对性能要求比较高的话就使用itemgetter() 方式。

  3.min(),max()

    最后,不要忘了这节中展示的技术也同样适用于min() 和max() 等函数。比如:
    >>> min(rows, key=itemgetter('uid'))
    {'fname': 'John', 'lname': 'Cleese', 'uid': 1001}
    >>> max(rows, key=itemgetter('uid'))
    {'fname': 'Big', 'lname': 'Jones', 'uid': 1004}