同时拥有 itemgetter 和 attrgetter答案

【问题标题】：Having itemgetter and attrgetter at the same time同时拥有 itemgetter 和 attrgetter
【发布时间】：2013-12-25 13:26:44
【问题描述】：

我正在尝试根据元素的属性在列表中选择最常见的项目。

a1 = Myclass(name='a')
a2 = Myclass(name='a')
b1 = Myclass(name='b')
l = [a1,a2,b1]

if most_common(l, attr_name='name') in [a1, a2]:
   # I want this true 
   # 'a' is the most occured 'name'

基本上我想修改代码https://stackoverflow.com/a/1520716/433570

key = operator.itemgetter(0) 然后 operator.attrgetter(attr_name)

想知道这是否可能？

【问题讨论】：

有什么理由不使用Counter？喜欢collections.Counter(x.name for x in l).most_common(1)

标签： python operator-keyword

【解决方案1】：

我偶然发现了这个问题，正在寻找同样的东西。我不喜欢接受的答案，因为它实际上并没有回答问题。看了 Python 的 attrgetter 实现，决定写一个。

https://docs.python.org/2/library/operator.html#operator.attrgetter

def resolve_itemattr(obj, attr):
    for name in attr.split("."):
        try:
            obj = obj[name]
        except TypeError:
            obj = getattr(obj, name)
    return obj


def itemattrgetter(attr):
    """
    Combo itemgetter and attrgetter
    """
    def g(obj):
        return resolve_itemattr(obj, attr)
    return g

它支持嵌套属性。我已经剥离了对多个 attrs 的处理，但将其放回原处并不难。

这让你使用 itemattrgetter 首先尝试获取项目，失败，获取 attr。

【讨论】：

【解决方案2】：

您可以预处理您的列表：

attr_getter = operator.attrgetter('name')
names = map(attr_getter, l)
if most_common(names) in set(map(attr_getter, [a1, a2])):
    pass # do smth here

【讨论】：