Наиболее эффективный способ вычитать между двумя списками в каждом общем ключе между двумя python словарями - PullRequest
Новая
       16

Наиболее эффективный способ вычитать между двумя списками в каждом общем ключе между двумя python словарями

0 голосов
/ 15 апреля 2020

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

Вот небольшой пример 

x={
    '1' : [1,2,3],
    '2' : [2,9]
}
y={
    '1' : [4,5],
    '3' : [8,9]
}

# Common key is '1' , so z is the list of the subtraction of elements of key '1'
>> expected_output = [-3, -2, -1, -4, -3, -2]
                   = [1-4,2-4,3-4, 1-5,2-5,3-5]

Вот пример настройки больших словарей со многими общими ключами и необычными ключами 

import random

x={}
y={}
num_same_keys = 100
num_diff_keys = 200

## Generating common keys with arbitrary number of elements
common_key = random.randint(0,10000)
for _ in range(num_same_keys) : 
    while common_key in x : common_key = random.randint(0,10000)
    x[common_key] = [ random.randint(0,1000) for _ in range(random.randint(1,10)) ]
    y[common_key] = [ random.randint(0,1000) for _ in range(random.randint(1,10)) ]

## Generating different keys with arbitrary number of elements
x_key = random.randint(0,10000)
y_key = random.randint(0,10000)
for _ in range(num_diff_keys) : 
    # Adding to x    
    while (x_key in x) and (x_key in y) : x_key = random.randint(0,10000)
    x[x_key] = [ random.randint(0,1000) for _ in range(random.randint(1,10)) ]
    # Adding to y
    while (y_key in x) and (y_key in y) : y_key = random.randint(0,10000)
    y[y_key] = [ random.randint(0,1000) for _ in range(random.randint(1,10)) ]

1 Ответ

0 голосов
/ 15 апреля 2020

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

def f1(x,y):
    z = []
    for element in x.keys() & y.keys():
        z.extend([xval-yval for xval in x[element] for yval in y[element] ])
    return z

outputlist = f1(x,y)
...