含有相同元素的两个列表转字典

含有相同元素的两个列表转字典

在这里插入代码片

第一种
from itertools import groupby
import numpy as np

aa = [0,1,2,2,0]
bb = [“小米”,“华为”,“联想”,“联想2”,“小米2”]
aa_sorted = sorted(aa) #[0, 0, 1, 2, 2]
index = sorted(range(len(aa)),key = lambda k:aa[k] ) #[0, 4, 1, 2, 3]
bb_sorted = np.array(bb)[index].tolist() #[‘小米’, ‘小米2’, ‘华为’, ‘联想’, ‘联想2’]
aa_sorted_binded = [list(j) for i,j in groupby(aa_sorted)] #[[0, 0], [1], [2, 2]]
index_new = [[0]]
index_new.append([len(ele) for ele in aa_sorted_binded]) #[[0], [2, 1, 2]]
index_new = sum(index_new,[]) #[0, 2, 1, 2]
bb_sorted_binded = [bb_sorted[sum(index_new[0:i+1]):sum(index_new[1:i+2])] for i in range(len(index_new)-1)] #[[‘小米’, ‘小米2’], [‘华为’], [‘联想’, ‘联想2’]]
result = dict(zip(list(set(aa)),bb_sorted_binded)) #{0: [‘小米’, ‘小米2’], 1: [‘华为’], 2: [‘联想’, ‘联想2’]}

第二种

bb=[“小米”,“华为”,“联想”,“联想2”,“小米2”]
aa=[0,1,2,2,0]
cc={}
count=0
for i in aa:
if i not in cc:
cc[i]=[bb[count]]
else:
cc[i].append(bb[count])
count+=1
cc