python - 同时拥有 itemgetter 和 attrgetter

Question

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

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)

想知道这是否可能？

Answer 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

它支持嵌套属性。我已经取消了对多个属性的处理，但将其放回去并不难。

这让您可以使用 itemattrgetter 首先尝试获取该项目，如果失败，则获取 attr。