operator的itemgetter和attrgetter

前几天在给个list做排序的时候，隐隐约约想起来有个语法糖可以替代lambda函数，用来获取list中dict的key，作为排序的key。

 

这个语法糖平时用得少，怎么都想不起来。今天查看python标准库operator时，终于把这两个方法找到了，做个笔记。

 

因为就是一个语法糖，太简单了，没必要做太多解释说明，直接贴代码吧。

 

from operator import itemgetter, attrgetter

__author__ = 'blackmatrix'


get_name = lambda user: user['name']

get_address = lambda user: getattr(user, 'address')


class User:

    def __init__(self, name, address):
        self.name = name
        self.address = address


laowang = {'name': '老王', 'address': '隔壁'}
laoli = User('老李', '楼下')


if __name__ == '__main__':
    # itemgetter('name') 和 get_name 作用是相同的
    # 都是接受一个对象，获取对象key为name的值
    # 主要用来替代只是为了获取value的lambda函数
    print(itemgetter('name')(laowang))
    print(get_name(laowang))

    # 同样的，attrgetter('address')也是等同于get_address
    # 用来替代只为了获取attr的lambda函数
    print(attrgetter('address')(laoli))
    print(get_address(laoli))