Блог

У меня есть код с двумя списками словарей, и я должен составить окончательный словарь на основе сравнения двух списков.

Это и есть код:

 list_a = [
   {'e1':20},
   {'e1':11.2},
   {'e1':20.33},
   {'e1':19.34},
   {'e1':18.2}
 ]

list_b = [
   {'id': 1, 'e1':20, 'status':True},
   {'id': 2, 'e1':11.2, 'status':False},
   {'id': 3, 'e1':20.33, 'status':True}
]

mydict = {}
for b in list_b:
  mydict.setdefault(b['id'], {})
  if b['status']:
    for a in list_a:
        if b['e1'] - a['e1'] > 1:
            mydict[b['id']] = b['e1'] - a['e1']
 

Просто интересно, есть ли у вас какое-либо решение, чтобы сделать этот код более эффективным (как в плане скорости, так и в плане пространства)?

Один из способов сократить время-избегать использования mydict.setdefault ,

 import time

list_a = [
    {'e1': 20},
    {'e1': 11.2},
    {'e1': 20.33},
    {'e1': 19.34},
    {'e1': 18.2}
]

list_b = [
    {'id': 1, 'e1': 20, 'status': True},
    {'id': 2, 'e1': 11.2, 'status': False},
    {'id': 3, 'e1': 20.33, 'status': True}
]

time_original = 0
time_new = 0

for loop in range(1000000):
    time1 = time.time()
    mydict = {}
    for b in list_b:
        mydict.setdefault(b['id'], {})
        if b['status']:
            for a in list_a:
                if b['e1'] - a['e1'] > 1:
                    mydict[b['id']] = b['e1'] - a['e1']
    time2 = time.time()

    time_original  = time2 - time1

    time3 = time.time()
    example_dict = {}
    for b in list_b:
        example_dict[b['id']] = {}
        if b['status']:
            for a in list_a:
                if b['e1'] - a['e1'] > 1:
                    example_dict[b['id']] = b['e1'] - a['e1']
    time4 = time.time()

    time_new  = time4 - time3

print('Using mydict.setdefault(b['id'], {})')
print(time_original / 1000000)
print(mydict)
print()
print('Using example_dict[b['id']] = {}')
print(time_new / 1000000)
print(example_dict)
 

Результаты,

 Using mydict.setdefault(b['id'], {})
2.1369829177856445e-06
{1: 1.8000000000000007, 2: {}, 3: 2.129999999999999}

Using example_dict[b['id']] = {}
2.096635103225708e-06
{1: 1.8000000000000007, 2: {}, 3: 2.129999999999999}