Неисправные словарные ключи после слияния в Python - PullRequest
Новая
       41

Неисправные словарные ключи после слияния в Python

1 голос
/ 24 сентября 2019

После объединения двух словарей подключи из полученного словаря являются беспорядочными.Подключами являются месяцы ['jan', 'feb', 'mar', ..., 'dec'].В некоторых случаях исходный словарь может не содержать подраздел (месяц), поэтому вывод будет дезориентирован при слиянии.

Так что у меня есть два словаря со следующей структурой {Model:{'Jan':[1], 'Feb':[2], Jun: [5], ...}

Какв этом примере вы можете видеть, что некоторые подразделы (месяцы) не представлены, поэтому они не могут быть найдены в исходных комментариях.Но мне нужен объединенный диктат для сохранения ежемесячного порядка, не имеет значения, как выглядели оригинальные диктанты.

Функция слияния: 

def merge_dicts(dict1, dict2):
    '''Marge two dicts by adding up (accumulating) values in each key.

    Returns: A merge (addition) of two dictionaries by adding values of same keys
    '''
   # Merge dictionaries and add values of same keys
    out = {**dict1, **dict2}
    for key, value in out.items():
        if key in dict1 and key in dict2:
            out[key] = [value, dict1[key]]
            #Things got harder, out[key] appends in list of list of list... no itertools can help here.
            lst = [] #the one dimensional list to fix the problem of list of list with out[key]
            for el in out[key]:
                try:
                #if inside out[key] there is a list of list we split it
                    for sub_el in el:
                        lst.append(sub_el)
                except:
                #if inside out[key] there is only a single float
                    lst.append(el)
            #Replace the old key with the one dimensional list
            out[key] = lst
    return out

Как я сливаю это:

for c in range(len([*CMs_selection.keys()])):
    if c == 0:
        #First merge, with dict0 & dict1
        merged_dict = {cm:merge_dicts(CMs_selection[[*CMs_selection.keys()][c]][cm], CMs_selection[[*CMs_selection.keys()][c + 1]][cm]) 
                       for cm in CMs_selection[[*CMs_selection.keys()][0]]}

    elif c > 0 and c < (len(years) - 1):
        #Second merge to n merge, starting with dict_merged and dict 2
        merged_dict = {cm:merge_dicts(merged_dict[cm], CMs_selection[[*CMs_selection.keys()][c + 1]][cm]) 
                       for cm in CMs_selection[[*CMs_selection.keys()][0]]}

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

{'Model1': {'Jan': [-0.0952586755156517,
   0.1015196293592453,
   -0.10572463274002075],
  'Oct': [-0.02473766915500164,
   0.0678798109292984,
   0.08870666474103928,
   -0.06378963589668274],
  'Nov': [-0.08730728179216385,
   0.013518977910280228,
   0.023245899006724358,
   -0.03917887806892395],
  'Jul': [-0.07940272241830826, -0.04912888631224632, -0.07454635202884674],
  'Dec': [-0.061335086822509766, -0.0033914903178811073, 0.09630533307790756],
  'Mar': [0.029064208269119263, 0.11327305436134338, 0.009556809440255165],
  'Apr': [-0.04433680325746536, -0.08620205521583557],
  'Jun': [-0.036688946187496185, 0.05543896555900574, -0.07162825018167496],
  'Aug': -0.03712410107254982,
  'Sep': [0.007421047426760197, 0.008665643632411957],
  'Feb': [-0.02879650704562664, 0.013025006279349327]},
 'Model2': {'Feb': -0.05173473060131073,
  'Jun': [-0.09126871824264526,
   -0.09009774029254913,
   0.10458160936832428,
   -0.09445420652627945,
   -0.04294373467564583],
  'Aug': [-0.07917020469903946, 0.011026041582226753],
  'Oct': [-0.10164830088615417, ....

....

С беспорядочными месяцами.Пожалуйста, помогите мне!

1 Ответ

0 голосов
/ 24 сентября 2019

Если мы просто сосредоточимся на слиянии словарей, сначала нам нужно определить нормальный порядок месяцев, а затем выполнить слияние в этом порядке, потому что Python не знает этого порядка.Он не может добавить «Mar» между «Feb» и «Apr», если он не существует в первом словаре.Итак, нам нужно определить порядок самостоятельно.

Кроме того, вам нужно два разных решения для объединения значений с плавающей запятой и слияния списков.Я добавил параметр mode в свое решение. 

def merge_dicts(list_of_dicts, mode):
    keys = set(key for d in list_of_dicts for key in d.keys())
    months = ["Jan","Feb","Mar","Apr","May","Jun","Jul","Aug","Sep","Oct","Nov","Dec"]
    ordered_keys = [month for month in months if month in keys]

    out = {}
    for key in ordered_keys:
        out[key] = []
        for d in list_of_dicts:
            if key in d:
                if mode == "append":
                    out[key].append(d[key]) 
                elif mode == "extend":
                    out[key].extend(d[key])

    return out


CMs_selection = {2006: {'Model1': {'Jan': -0.1, 'Oct': -0.063}, 'Model2': {'Feb': -0.051, 'Jun': -0.04, 'Oct': 0.07}, 'Model3': {'Mar': -0.030, 'Jun': 0.02, 'Aug': 0.0561,}, 'Model4': {'Feb': -0.026, 'Dec': -0.06}}, 2007: {'Model1': {'Jul': -0.07, 'Oct': 0.8, 'Nov': 0.38, 'Dec': 0.1}, 'Model2': {'Jun': -0.09, 'Aug': -0.079, 'Sep': -0.7}}}

for key in CMs_selection:
    CMs_selection[key] = merge_dicts(CMs_selection[key].values(), "append")

print(CMs_selection)

result = merge_dicts(CMs_selection.values(), "extend")
print(result)

Вывод: 

{2006: {'Jan': [-0.1], 'Feb': [-0.051, -0.026], 'Mar': [-0.03], 'Jun': [-0.04, 0.02], 'Aug': [0.0561], 'Oct': [-0.063, 0.07], 'Dec': [-0.06]}, 2007: {'Jun': [-0.09], 'Jul': [-0.07], 'Aug': [-0.079], 'Sep': [-0.7], 'Oct': [0.8], 'Nov': [0.38], 'Dec': [0.1]}}
{'Jan': [-0.1], 'Feb': [-0.051, -0.026], 'Mar': [-0.03], 'Jun': [-0.04, 0.02, -0.09], 'Jul': [-0.07], 'Aug': [0.0561, -0.079], 'Sep': [-0.7], 'Oct': [-0.063, 0.07, 0.8], 'Nov': [0.38], 'Dec': [-0.06, 0.1]}
...