Python：将元组/字典作为键，进行选择，排序

Question

假设我有大量不同颜色的水果，例如24个蓝色香蕉，12个绿色苹果，0个蓝色草莓等等。我想将它们组织成Python的数据结构，以便于选择和排序。我的想法是将它们放入以元组为键的字典中，例如，

{
    ('banana',    'blue' ): 24,
    ('apple',     'green'): 12,
    ('strawberry','blue' ): 0,
    # ...
}

甚至字典，例如

{
    {'fruit': 'banana',    'color': 'blue' }: 24,
    {'fruit': 'apple',     'color': 'green'}: 12,
    {'fruit': 'strawberry','color': 'blue' }: 0,
    # ...
}

例如，我想检索所有蓝色水果或所有颜色的香蕉的列表，或按水果名称对字典进行排序。有没有办法做到这一点？

用元组作为键的字典可能不是处理这种情况的 正确 方法。

欢迎所有建议！

Answer 1

就个人而言，我最喜欢python的一件事是tuple-dict组合。您在这里拥有的实际上是一个2d数组（其中x =水果名称，y
=颜色），而且我通常是实现2d数组的元组字典的支持者，至少在诸如之类numpy的数据库不适合使用时。简而言之，我认为您有一个很好的方法。

请注意，如果不做一些额外的工作，就不能将字典用作字典中的键，因此这不是一个很好的解决方案。

也就是说，您还应该考虑namedtuple（）。这样，您可以执行以下操作：

>>> from collections import namedtuple
>>> Fruit = namedtuple("Fruit", ["name", "color"])
>>> f = Fruit(name="banana", color="red")
>>> print f
Fruit(name='banana', color='red')
>>> f.name
'banana'
>>> f.color
'red'

现在您可以使用fruitcount字典：

>>> fruitcount = {Fruit("banana", "red"):5}
>>> fruitcount[f]
5

其他技巧：

>>> fruits = fruitcount.keys()
>>> fruits.sort()
>>> print fruits
[Fruit(name='apple', color='green'), 
 Fruit(name='apple', color='red'), 
 Fruit(name='banana', color='blue'), 
 Fruit(name='strawberry', color='blue')]
>>> fruits.sort(key=lambda x:x.color)
>>> print fruits
[Fruit(name='banana', color='blue'), 
 Fruit(name='strawberry', color='blue'), 
 Fruit(name='apple', color='green'), 
 Fruit(name='apple', color='red')]

与chmullig相呼应，要获得一个水果的所有颜色的列表，您必须过滤键，即

bananas = [fruit for fruit in fruits if fruit.name=='banana']