根据值将Python中的两个列表列表分组

Question

我在Python中有两个清单。

list1 = ['a','a','b','a','c','b','c','a','d','a','b']
list2 = ['1','2','21','12','1','32','11','12','21','3','31']

我必须将list1的相似元素分组。 list2的相应元素也应基于此进行分组。 输出应为：

list1 = [['a','a','a','a','a'],['b','b','b'],['c','c'],['d']]
list2 = [['1','2','12','12','3'],['21','32','31'],['1','11'],['21']]

做这个的最好方式是什么？

Answer 1

如果您不关心第一个列表中元素的顺序，则可以使用defaultdict ：

In [7]: from collections import defaultdict

In [8]: from itertools import izip

In [9]: res = defaultdict(list)

In [10]: for k, v in izip(list1, list2):
   ....:     res[k].append(v)
   ....:     

In [11]: print(res)
defaultdict(<type 'list'>, {'a': ['1', '2', '12', '12', '3'], 'c': ['1', '11'], 'b': ['21', '32', '31'], 'd': ['21']})

In [12]: res.items()
Out[12]: 
[('a', ['1', '2', '12', '12', '3']),
 ('c', ['1', '11']),
 ('b', ['21', '32', '31']),
 ('d', ['21'])]

Answer 2

此代码应执行以下操作：

final_list1 = []
final_list2 = []

for distinct in sorted(list(set(list1))):
    index = 0
    distinct_list1 = []
    distinct_list2 = []
    for element in list1:
        if element == distinct:
            distinct_list1.append(element)
            distinct_list2.append(list2[index])
        index += 1
    final_list1.append(distinct_list1)
    final_list2.append(distinct_list2)

list1 = final_list1
list2 = final_list2

这将为您提供所需的确切输出。 如果您不太在乎输出，那么@soon建议可能有更好的方法。

Answer 3

这段代码对我有用：

groups = list(set(list1))
list1_tmp, list2_tmp = [], []
for char in groups:
   list1_tmp.append([])
   list2_tmp.append([])

for i in range(len(list1)):
   list1_tmp[groups.index(list1[i])].append(list1[i])
   list2_tmp[groups.index(list1[i])].append(list2[i])

list1 = list1_tmp
list2 = list2_tmp

该输出对于任何其他类似输入也应有效。

Answer 4

这是一个（有点丑陋的）实现，可以解决这个问题：

list1 = ['a','a','b','a','c','b','c','a','d','a','b']
list2 = ['1','2','21','12','1','32','11','12','21','3','31']

def transform(in_list, other_list):
    if len(in_list) != len(other_list):
        raise ValueError("Lists must have the sema length!")
    out_list = list()
    out_other_list = list()
    for i, c in enumerate(in_list):
        for inner_list, inner_other_list in zip(out_list, out_other_list):
            if c in inner_list:
                inner_list.append(c)
                inner_other_list.append(other_list[i])
                break
        else:
            out_list.append([c])
            out_other_list.append([other_list[i]])
    return out_list, out_other_list

print transform(list1, list2)

Answer 5

尽管我个人希望很快得到答复 ，但此人成功检索了所需的输出。

lst= sorted(zip(list1,list2),key=lambda x:x[0])

intList=[]

initial=lst[0][0]
count=0

for index,value in enumerate(lst):
    if value[0]==initial:
        continue
    else:
        intList.append(lst[count:index])
        initial=value[0]
        count=index

finList1=[[a for a,b in innerList] for innerList in intList]
finList2=[[b for a,b in innerList] for innerList in intList]

Answer 6

第一个案例可以使用itertools.groupby分组：

groups = list()
for k, group in itertools.groupby(list1):
    groups.append(list(group))

第二个类似地基于您的keyfunc。 阅读有关itertools.groupby的更多信息