如何过滤餐厅字典列表并调整键值对输出顺序？

调用get_rest(rest)函数后得到rest_list变量，内容如下：

[{'close_time': '5:00 pm',
  'open_time': '8:00 am',
  'name': 'Pizza DeVille',
  'city': 'Las Vegas',
  'availability': 'available'},
 {'close_time': '10:00 pm',
  'open_time': '9:00 am',
  'name': 'Cecil’s',
  'city': 'Philadelphia',
  'availability': 'available'}]

需要实现两个需求：

1. 过滤出仅包含availability值为'available'的字典
2. 调整字典键的顺序，将'availability'置于首位，示例输出如下：

[{'availability': 'available', 'close_time': '5:00 pm', 'city': 'Las Vegas', 'open_time': '8:00 am', 'name': 'Pizza DeVille'}, {'availability': 'available', 'close_time': '10:00 pm', 'city': 'Philadelphia', 'open_time': '9:00 am', 'name': 'Cecil’s'}]

现有filter_rest函数缺少对应逻辑，代码如下：

def filter_rest(rest):
    rest_list = get_rest(rest) #function that gives me first code chunk above
    new_list = []
    for x in rest_list:
        new_list.append(x)
    return new_list

1. 过滤逻辑实现 #

遍历餐厅列表时，通过判断字典中availability的值是否为'available'，筛选出符合条件的条目。用dict.get()方法可以避免字典没有availability键时抛出异常。

2. 调整键顺序 #

Python 3.7及以上版本的字典会保留键的插入顺序，因此可以重新构造字典：先放入availability键值对，再依次添加剩余的键值对（跳过已添加的availability）。

完整修改代码 #

def filter_rest(rest):
    rest_list = get_rest(rest)
    new_list = []
    for restaurant in rest_list:
        # 过滤出availability为available的餐厅
        if restaurant.get('availability') == 'available':
            # 构造新字典，将availability放在首位
            ordered_rest = {'availability': restaurant['availability']}
            # 添加剩余键值对
            for key, val in restaurant.items():
                if key != 'availability':
                    ordered_rest[key] = val
            new_list.append(ordered_rest)
    return new_list

简化写法（字典推导式） #

如果想让代码更简洁，可以用字典解包的方式合并键值对：

def filter_rest(rest):
    rest_list = get_rest(rest)
    new_list = []
    for restaurant in rest_list:
        if restaurant.get('availability') == 'available':
            ordered_rest = {
                'availability': restaurant['availability'],
                **{k: v for k, v in restaurant.items() if k != 'availability'}
            }
            new_list.append(ordered_rest)
    return new_list

内容的提问来源于stack exchange，提问作者emvirgen