用Python中的字典循环遍历字典列表的正确方法

Question

我有一份车辆清单和一份过滤器清单。代码如下：

过滤器字典如下：

filters = {
  'make': 'SKODA',
  'model': 'Fabia',
  'fuel': 'Diesel',
  'transmission': '',
  'registration_year': '',
  'price': {'start': 10000, 'stop': 12000}
}

车辆清单如下：

vehicles = [
  {'make': 'AUDI', 'model': 'Q2 Dsl', 'type': '1.6 TDi Sport', 'fuel': 'Diesel', 'mileage': '19896', 'registration_year': '2017', 'transmission': 'Handbediende versnellingsbak', 'price': 17800}, 
  {'make': 'AUDI', 'model': 'A6 SW Dsl', 'type': '2.0 TDi S line', 'fuel': 'Diesel', 'mileage': '87354', 'registration_year': '2013', 'transmission': 'Handbediende versnellingsbak', 'price': 52000},
  {'make': 'SKODA', 'model': 'Fabia', 'type': '1.6 CR TDi GreenLine Active DPF', 'fuel': 'Diesel', 'mileage': '90613', 'registration_year': '2012', 'transmission': 'Handbediende versnellingsbak', 'price': 11000},
  {'make': 'AUDI', 'model': 'A4 SW Dsl', 'type': '2.0 TDi S tronic', 'fuel': 'Diesel', 'mileage': '47402', 'registration_year': '2016', 'transmission': 'Sequentiele bak', 'price': 93000},
  {'make': 'VOLKSWAGEN', 'model': 'Touran', 'type': '1.4 TSI Trendline', 'fuel': 'Essence', 'mileage': '28588', 'registration_year': '2017', 'transmission': 'Handbediende versnellingsbak', 'price': 87000},
  {'make': 'AUDI', 'model': 'A4 Dsl', 'type': '2.0 TDi', 'fuel': 'Diesel', 'mileage': '66053', 'registration_year': '2014', 'transmission': 'Handbediende versnellingsbak', 'price': 62000}
]

因此，我想在所有车辆和所有过滤器之间循环，只返回那些与过滤器匹配的车辆。在这种情况下，我应该只得到斯柯达。你知道吗

我尝试了pydash库，但没有成功。你知道吗

我试过的是：

def filter_item(vehicle):
  vehicles = []
  for key, value in filters.items():
    if key == "price":
      if vehicle[key] > value['start'] and vehicle[key] < value['stop']:
        vehicles.append(vehicle)
    else:
      if vehicle[key] == value:
        vehicles.append(vehicle)
  return vehicles

result = py_.filter(vehicles, lambda x: filter_item(x))

但在这种情况下，我得到的是所有燃油类型为柴油的车辆，但如果从0开始，我应该只得到第二辆，因此只有斯柯达。

你知道我该怎么做吗？有没有更好的方法或更干净的代码来做到这一点？你知道吗

Answer 1

您需要检查所有的过滤条件，这里有一个python内置的例子filter：

def filter_item(vehicle):
    vehicles = []
    for key, value in filters.items():
        if not value: # skip empty filter
            continue
        if key == "price":
            if vehicle[key] > value['start'] and vehicle[key] < value['stop']:
                continue
        else:
            if vehicle[key] == value:
                continue
        return False # no filter matched
    return True


result = filter(filter_item, vehicles)

使用pydash：

result = py_.filter(vehicles, filter_item)

注意：传递给fitler的回调函数返回true，以指示当前值应包含在结果中。你知道吗

Answer 2

我认为价格范围应该从filters分开。这是尴尬的，因为它是唯一的关键，你不能直接匹配相应的值为同一辆车的关键。你知道吗

我会这样写

>>> filters = {
...:  'make': 'SKODA',
...:  'model': 'Fabia',
...:  'fuel': 'Diesel',
...:  'transmission': '',
...:  'registration_year': '',
...:}
>>> price_range = range(10000, 12000 + 1)
>>>
>>> [vehicle['make'] for vehicle in vehicles if
...:all(vehicle.get(k) == v for k, v in filters.items() if v) and
...:vehicle['price'] in price_range]
>>> ['SKODA']

或者像这样

>>> [vehicle for vehicle in vehicles if
...:all(vehicle.get(k) == v for k, v in filters.items() if v) and
...:vehicle['price'] in price_range]
>>> 
[{'fuel': 'Diesel',
  'make': 'SKODA',
  'mileage': '90613',
  'model': 'Fabia',
  'price': 11000,
  'registration_year': '2012',
  'transmission': 'Handbediende versnellingsbak',
  'type': '1.6 CR TDi GreenLine Active DPF'}]

以防您需要匹配的整车词典。你知道吗