Генерировать список значений на основе совпадающих ключей

#python-3.7 #dictionary-comprehension

#python-3.7 #понимание по словарю

Вопрос:

У меня есть приведенный ниже словарь:

 {'Closed': {'High': 33, 'Medium': 474, 'Low': 47, 'Critical': 6}, 'Impact Statement Pending': {'Low': 3, 'Medium': 1, 'Critical': 0, 'High': 0}, 'New': {'Low': 1, 'High': 2, 'Critical': 2, 'Medium': 2}, 'Remediation Plan Pending': {'Medium': 10, 'Low': 1, 'Critical': 1, 'High': 0}, 'Remedy in Progress': {'Medium': 36, 'Low': 18, 'High': 4, 'Critical': 1}}
  

Как я мог бы создать список, состоящий из всех значений для указанного ключа? Список для всех высоких значений или другой список для всех средних значений?

Способ, которым я в настоящее время добиваюсь этого, не кажется лучшим способом. У меня есть список всех уровней серьезности, который я повторяю и сравниваю, как показано ниже:

 trace_list = ['High', 'Medium', 'Critical', 'Low']

total_status_dict = {'Closed': {'High': 33, 'Medium': 474, 'Low': 47, 'Critical': 6}, 'Impact Statement Pending': {'Low': 3, 'Medium': 1, 'Critical': 0, 'High': 0}, 'New': {'Low': 1, 'High': 2, 'Critical': 2, 'Medium': 2}, 'Remediation Plan Pending': {'Medium': 10, 'Low': 1, 'Critical': 1, 'High': 0}, 'Remedy in Progress': {'Medium': 36, 'Low': 18, 'High': 4, 'Critical': 1}}

for item in trace_labels:

     y_values = []

     for key, val in total_status_dict.items():
          for ke in total_status_dict[key]:
               if item is ke:
                    y_values.append(total_status_dict[key][ke])
  

Ответ №1:

Примечание: вы перебираете total_status_dict ключи и добавляете результаты в список. Помните, что даже если словари официально упорядочены в Python начиная с 3.7 (см. https://docs.python.org/3/whatsnew/3.7.html ) вы не всегда контролируете версию пользователя на Python. Я бы предпочел создать dict key -> item -> value , где key есть Closed , Impact Statement Pending , … и item является одним из trace_labels , чем dict, key -> [values] где values предполагается порядок, как в trace_labels .

Ваш код неэффективен, потому что вы повторяете trace_labels дважды:

  • for item in trace_labels:
  • for ke in total_status_dict[ключ]: если элемент равен ke:`

Как выполнить итерацию только один раз? Вместо построения y_values списков один за другим (с повторением всей итерации total_status_dict каждый раз), вы можете создать несколько списков одновременно:

 >>> trace_labels = ['High', 'Medium', 'Critical', 'Low']
>>> total_status_dict = {'Closed': {'High': 33, 'Medium': 474, 'Low': 47, 'Critical': 6}, 'Impact Statement Pending': {'Low': 3, 'Medium': 1, 'Critical': 0, 'High': 0}, 'New': {'Low': 1, 'High': 2, 'Critical': 2, 'Medium': 2}, 'Remediation Plan Pending': {'Medium': 10, 'Low': 1, 'Critical': 1, 'High': 0}, 'Remedy in Progress': {'Medium': 36, 'Low': 18, 'High': 4, 'Critical': 1}}
>>> y_values_by_label = {}
>>> for key, value_by_label in total_status_dict.items():
...     for label, value in value_by_label.items(): # total_status_dict[key] is value_by_label
...         y_values_by_label.setdefault(label, {})[key] = value
...
>>> y_values_by_label
{'High': {'Closed': 33, 'Impact Statement Pending': 0, 'New': 2, 'Remediation Plan Pending': 0, 'Remedy in Progress': 4}, 'Medium': {'Closed': 474, 'Impact Statement Pending': 1, 'New': 2, 'Remediation Plan Pending': 10, 'Remedy in Progress': 36}, 'Low': {'Closed': 47, 'Impact Statement Pending': 3, 'New': 1, 'Remediation Plan Pending': 1, 'Remedy in Progress': 18}, 'Critical': {'Closed': 6, 'Impact Statement Pending': 0, 'New': 2, 'Remediation Plan Pending': 1, 'Remedy in Progress': 1}}
  

setdefault(label, {}) создает пустой dict, y_values_by_label[label] = {} если y_values_by_label у него нет ключа label .

Если вы хотите превратить это в понимание dict, вам придется использовать свой неэффективный метод:

 >>> {label:{k:v for k, value_by_label in total_status_dict.items() for l, v in value_by_label.items() if l==label} for label in trace_labels}
{'High': {'Closed': 33, 'Impact Statement Pending': 0, 'New': 2, 'Remediation Plan Pending': 0, 'Remedy in Progress': 4}, 'Medium': {'Closed': 474, 'Impact Statement Pending': 1, 'New': 2, 'Remediation Plan Pending': 10, 'Remedy in Progress': 36}, 'Critical': {'Closed': 6, 'Impact Statement Pending': 0, 'New': 2, 'Remediation Plan Pending': 1, 'Remedy in Progress': 1}, 'Low': {'Closed': 47, 'Impact Statement Pending': 3, 'New': 1, 'Remediation Plan Pending': 1, 'Remedy in Progress': 18}}