python - 对具有多个大小值的字典进行排序

Question

我有一本字典需要排序。我知道字典不能排序，所以我正在构建一个列表来根据索引执行排序。我遇到的问题是多个键之间存在重复值；此外，值可以是多个列表。

字典:

my_dict = {
    '1' : ['Red', 'McDonald\'s'],
    '2' : [['Orange', 'Wendy\'s'], ['Purple', 'Cookout']],
    '3' : ['Yellow', 'Longhorn'],
    '4' : ['Green', 'Subway'],
    '5' : ['Blue', 'Chipotle'],
    '6' : ['Indigo', 'Taco Bell'],
    '7' : [['Violet', 'Steak n Shake'], ['Dark Red', 'Five Guys']],
    '8' : ['Dark Orange', 'Wendy\'s'],
    '9' : ['Aqua', 'Firehouse Subs'],
    '10' : ['Magenta', 'McDonald\'s'],
}

我需要能够按键(数字)、颜色和餐厅对其进行排序。我遇到的问题是当有重复时打印必须分组，我无法弄清楚如何正确排序我生成的列表。

按数字打印示例:

1:
    Red, McDonald's

2:
    Orange, Wendy's
    Purple, Cookout

3:
    Yellow, Longhorn

...

按颜色打印的示例:

Aqua:
    Firehouse Subs, 9

Blue:
    Chipotle, 5

Dark Orange:
    Wendy's, 8

...

我的解决方案是创建一个列表，其中包含列表中的每个键和值。我还是新手，我认为这不是迭代该字典的正确方法，但它适用于给定的字典。

sorted_number = []
for key, value in my_dict.items():
    if len(value[0][0]) > 1:
        working_list = (key, value[0][0], value[0][1], value[1][0], value[1][1])
        sorted_number.append(working_list)
    else:
        working_list = (key, value[0], value[1])
        sorted_number.append(working_list)

sorted_years = sorted(sorted_years, key=lambda y: y[0][0])
for i in sorted_years:
    print('{}:'.format(i[0]))
    if len(i) > 3:
         print('\t{}, {}'.format(i[1], i[2])
         print('\t{}, {}'.format(i[3], i[4]))
    else:
         print('\t{}, {}'.format(i[1], i[2]))

因为有些值是多个列表，if/else 语句确定该值是一个还是两个列表。然后，这将正确生成一个列表(数字、颜色、餐厅)或(数字、颜色、餐厅、颜色、餐厅)，然后可以按数字对列表进行排序。问题是当我尝试按颜色或餐厅排序时。我知道我可以让列表只有三个元素来使排序正常工作，但是当我去打印时，我最终会得到多个 2 和 7。餐厅的结果相同。此外，这并没有解决多家餐厅的问题。我已经搜索和测试了大约五天，尝试用不同的方法解压字典，但这是我最接近解决这个问题的方法。任何帮助表示赞赏。提前致谢。

为清楚起见，这里是数字的打印迭代:

for i in sorted_number:
    print('{}:'.format(i[0]))
    if len(i) > 3:
        print('\t{}, {}'.format(i[1], i[2]))
        print('\t{}, {}'.format(i[3], i[4]))
    else:
        print('\t{}, {}'.format(i[1], i[2]))

Answer 1

第一步是将dict转化为list，将dict的值均质化为元组数组的形式。然后使用 groupBy itertools 模块中的函数，您可以按所选参数分组。

class RestourantSorter:
    def __init__(self, source):
        self._rList = []
        """ Dict to list [(key, color, restourant name), ....] """
        for key, value in source.items():
            for color, rest in value if isinstance(value[0], list) else [value]:
                self._rList.append((key,color,rest))

    def _byIndex(self, x):
        return int(x[0])
    def _byColor(self, x):
        return x[1]
    def _byName(self, x):
        return x[2]

    def sort(self, sortFn):
        """ Sorted data [(key, [... values ...]), ....] """
        groups = []
        data = sorted(self._rList, key=sortFn)
        for k, g in groupby(data, sortFn):
            groups.append((k, list(g)))
        return groups

    def sortByIndex(self):
        return self.sort(self._byIndex)
    def sortByColor(self):
        return self.sort(self._byColor)
    def sortByName(self):
        return self.sort(self._byName)

工作 REPL