Как обьединить 2 списка в 1 словарь на Python?

Question

[Владимир Филинский]

Есть 2 списка:
ID_Masterok = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
ID_Master = ['Интендификатор мастера', '654', '2425', '654', '654', '—', '—', '—', '—', '—', '—']

Выполняя:
ID_Mastework = dict(zip(ID_Master, ID_Masterok))
Выдает: {'Интендификатор мастера': 0, '654': 4, '2425': 2, '—': 10}

Мне нужно получить что-то такое: {'Интендификатор мастера': 0, '654': [1, 3, 4], '2425': 2, '—': [7, 8, 9, 10]}

Comments

[Ivan Yakushenko]

И по какому принципу производится слияние? Логики же вообще нет.

[Сергей Горностаев]

Ivan Yakushenko, по щучьему велению :)

[Влад Григорьев]

Владимир Филинский, если нужна специфичная логика то проще через обычный цикл или ковыряйте группировку

ID_Masterok = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
ID_Master = ['Интендификатор мастера', '654', '2425', '654', '654', '—', '—', '—', '—', '—', '—']

result = {}
for k, v in zip(ID_Master, ID_Masterok):
    if k not in result:
        result[k] = v
    elif isinstance(result[k], list):
        result[k].append(v)
    else:
        result[k] = [result[k], v]

print(result)

[o5a]

Владимир Филинский,
Лучше придерживаться одного формата данных, не придется в дальнейшем проверять тип и по-разному обрабатывать его. Т.е. даже в случае одиночного значения все равно использовать список, в данном примере вместо 0 использовать [0].

{'Интендификатор мастера': 0, '654': [1, 3, 4], '2425': 2, '—': [5, 6, 7, 8, 9, 10]}
{'Интендификатор мастера': [0], '654': [1, 3, 4], '2425': 2, '—': [5, 6, 7, 8, 9, 10]}

Answer 1

[Drill]

def mastework(id_key, id_value):
    id_join = {}
    for k, v in zip(id_key, id_value):
        id_join[k] = id_join.get(k, []) + [v]
    return {k:v if len(v) > 1 else v[0] for k, v in id_join.items()}

Пример:

ID_Masterok = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
ID_Master = ['Интендификатор мастера', '654', '2425', '654', '654', '—', '—', '—', '—', '—', '—']

ID_Mastework = mastework(ID_Master, ID_Masterok)
print(ID_Mastework)

In [55]:
{'Интендификатор мастера': 0, '654': [1, 3, 4], '2425': 2, '—': [5, 6, 7, 8, 9, 10]}

Comments

[Владимир Филинский]

Спасибище огромное!

[Drill]

Владимир Филинский,
За спасибо - спасибо. Но ответ можно отметить решением.

Answer 2

[aRegius]

from collections import defaultdict
dict_result = defaultdict(list)
for index, key in enumerate(id_master):
        dict_result[key].append(index)

Answer 3

[Александр]

не видел такой функции, проще написать руками:

def zzip(ID_Master,ID_Masterok):
	result = {}
	for i in ID_Masterok:
		if ID_Master[i] in result:
			if isinstance(result[ID_Master[i]],list):
				result[ID_Master[i]].append(i)
			else:
				result[ID_Master[i]] = [result[ID_Master[i]],i]
		else:
			result[ID_Master[i]] = i
	return result

Comments

[Владимир Филинский]

Я может тупой, но у меня не выходит истользовать.
Делал так:

ID_Masterwork = []
def zzip(ID_Master,ID_Masterok):

  for i in ID_Masterok:
    if ID_Master[i] in result:
      if isinstance(result[ID_Master[i]], list):
        result[ID_Master[i]].append(i)
      else:
        result[ID_Master[i]] = [result[ID_Master[i]], i]
    else:
      result[ID_Master[i]] = i
  return result

ID_Mastework = zzip(ID_Master, ID_Masterok)
print("ID_Masterwork= ", ID_Masterwork)

Выдает:
ID_Masterwork= []

[Александр]

ID_Mastework = zzip(ID_Master, ID_Masterok)

пропущена буква r