按对象属性访问Python对象列表

时间:2015-02-20 01:48:16

标签: python

我不知道我试图做的是不是Pythonic,我只是想做错,或者如果我不知道如何正确地提出这个问题。能够做到这一点对我来说是有意义的,但我已经搜索了15种不同的方法,但找不到答案。

我想做的事情似乎很简单:我有一个对象列表。我想通过对象的属性访问该列表。此代码有效:

class Fruit:
    def __init__(self, name, color):
        self.name = name
        self.color = color

    def __str__(self):
        return self.name    

class BaseballPlayer:
    def __init__(self, name, number, position):
        self.name = name
        self.number = number
        self.position = position

    def __str__(self):
        return self.name    

class ListByProperty(list):
    def __init__(self, property, list):
        super(ListByProperty, self).__init__(list)
        self.property = property

    def __getitem__(self, key):
        return [item for item in self if getattr(item, self.property) == key][0]

fruits = ListByProperty('name', [Fruit('apple', 'red'), Fruit('banana', 'yellow')])

baseballTeam = ListByProperty('position', [BaseballPlayer('Greg Maddux', 31, 'P'),
                                           BaseballPlayer('Javy Lopez', 8, 'C')])
teamByNumber = ListByProperty('number', baseballTeam)

print 'Apples are', fruits['apple'].color

pitcher = baseballTeam['P']
print 'The pitcher is #%s, %s' % (pitcher.number, pitcher)
print '#8 is', teamByNumber[8]

>>> Apples are red
The pitcher is #31, Greg Maddux
#8 is Javy Lopez

但是我真的必须自己创建列表来做这么简单的事情吗?除了循环或listcomp之外,没有通用的方法吗?这似乎应该是一个非常常见的事情,通过对象的属性获得列表中的对象和访问项。看起来它应该以类似于sorted(key=...)的方式得到普遍支持。

请注意,这与需要dict的情况不同。事实上,使用对象列表而不是字典的重点是避免做类似的事情:

fruits = {'apple': Fruit('apple', 'red')}

...要求您输入apple两次。似乎应该有一种通用的方法来做这样的事情:

print 'Apples are', fruits['apple'].color

......无需继承list

好吧,你可以建立一个像这样的字典:

fruits = [Fruit('apple', 'red'), Fruit('banana', 'yellow')]
fruits = {f.name: f for f in fruits}

或者你可以单行,但看起来仍然......呃......语法上很酸? :)

到目前为止,我发现的最好方法是:

class DictByProperty(dict):
    def __init__(self, property, list):
        super(DictByProperty, self).__init__({getattr(i, property): i for i in list})
        self.property = property

fruits = DictByProperty('name', [Fruit('apple', 'red')])

哦,好的,谢谢,我已经从这个问题中学到了很多东西。

2 个答案:

答案 0 :(得分:4)

class Fruit:
    def __init__(self, name, color):
        self.name = name
        self.color = color

fruits = dict(zip(['apple', 'banana'], [Fruit('apple', 'red'), Fruit('banana', 'yellow')]))

print("An apple is %s" % fruits['apple'].color)

OR:

fruits = {fruit.name : fruit for fruit in [Fruit('apple', 'red'), Fruit('banana', 'yellow')]}

print("An apple is %s" % fruits['apple'].color)

以下确实产生了一个集合:

fruits = {Fruit('apple', 'red'), Fruit('banana', 'yellow')}

请注意与我创建dict的方式不同

fruits = [Fruit('apple', 'red'), Fruit('banana', 'yellow')]

print("An apple is %s" % fruits[fruits.index('apple')].color)

不起作用,因为你的列表包含水果类型的对象而不是字符串,这就是同样的故事:

fruits = FruitList([Fruit('apple', 'red'), Fruit('banana', 'yellow')])

print("An apple is %s" % fruits['apple'].color)

要使上述方法有效,请执行以下操作:

class Fruit:
    def __init__(self, name, color):
        self.name = name
        self.color = color

    def __eq__(self, other):
        if isinstance(other, Fruit):
            return (self.name, self.color) == (other.name, other.color)
        return self.name == other

fruits = [Fruit('apple', 'red'), Fruit('banana', 'yellow')]
print("An apple is %s" % fruits[fruits.index('apple')].color)

我不建议这样做,因为如果您的列表恰好包含字符串apple,则对color的属性调用将变为无效,因为字符串没有名为{{的属性1}}

答案 1 :(得分:0)

您可以创建一个实例。

apple = Fruit('apple', 'red')
print(apple.color) # I use Python 3.x

我不确定我是否在关注你的问题。但也许这有帮助。

编辑:试图恢复我的声誉......

你可以在字典中使用WITHIN实例。例如......

banana = Fruit('banana', 'yellow')
apple = Fruit('apple', 'red')
fruits = {'apple':apple, 'banana':banana}

或者如果您愿意:

fruits = {'apple':Fruit('apple', 'red'), 'banana':Fruit('banana', 'yellow')}

无论哪种方式,您都可以使用

调用水果的颜色
print(fruits['banana'].color)