如何在python中将嵌套列表排序为具有唯一值的单独列表？

Question

我有两个变量：unique_val = [1,2,3]nested_list = [['name1',1],['name2',1],['name3',3],['name4',2],['name5',2],['name6',3]]基本上，我希望每个唯一值都有单独的名称列表。我努力拼凑了一组嵌套的for循环，但无济于事。理想情况下，输出将是这样的：list_1 = ['name1','name2']list_2 = ['name4','name5']list_3 = ['name3',name6']

Answer 1

为其中的每个项目创建变量unique_val不是一个好主意。与其进行硬编码，不如将所有字典更好地与键配合使用，list_1因为它可以处理任意数量的变量。

>>> from collections import defaultdict

>>> dic = defaultdict(list)

>>> nested_list = [['name1',1],['name2',1],['name3',3],['name4',2],['name5',2],['name6',3]]

>>> unique_val = [1,2,3]     #better make this a set to get O(1) lookup

>>> for v,k in nested_list:

        if k in unique_val:

            dic['list_'+str(k)].append(v)

#now access those lists:

>>> dic['list_1']

['name1', 'name2']

>>> dic['list_2']

['name4', 'name5']

>>> dic['list_3']

['name3', 'name6']

如果您有一个4，unique_val 那么您可能希望list_4成为一个空列表，这很容易由来处理defaultdict：

>>> dic['list_4']

[]

Answer 2

您可以使用列表理解：

[[name for name, number in nested_list if number == n] for n in unique_val]

如果您确实要将其放在单独的变量中，则可以执行以下操作。

list_1, list_2, list_3 = [[name for name, number in nested_list if number == n] for n in unique_val]