Question

我有一个嵌套列表，看起来像这样：

first_list = [[a, 1], [b, 3], [a, 6], [a, 2], [b, 4], [b, 5], ...]

我想按它们的第一个元素对它们进行分组，并创建一个新的嵌套列表，如下所示：

new_list = [ [1, 6, 2, ...], [3, 4, 5, ...], ...]

所有以go开头的元素都位于第一个子列表中，依此类推。在运行时之前不知道不同值a，b等的数量，或者我可以做类似的事情：

a_list = []
b_list = []
for tag, x in first_list:
    if tag == a:
        a_list.append(x)
    elif tag == b:
        b_list.append(x)
new_list = [a_list, b_list]

但是，我正在努力使它适应任意数量的标签。

我可能省略了问题的重要部分，但我应该说我已经有一个“标签”列表，即：

tags = [a, b, c, d, ...]

它们实际上不是字符，因此没有反逗号，但是无论如何它们都应该是可哈希的。

Answer 1

首先，您的a和b应该应该是字符串。

您可以使用列表推导来做到这一点：

first_list = [["a", 1], ["b", 3], ["a", 6], ["a", 2], ["b", 4], ["b", 5]]
a_list = [x for x in first_list if x[0] == "a"]
b_list = [x for x in first_list if x[0] == "b"]
new_list = [a_list, b_list]

Answer 2

使用Python和一般的编程，应该避免创建数量可变的变量。

defaultdict

您可以使用defaultdict个对象中的list个。这自然可以扩展到任意数量的组，而不必显式命名变量。

first_list = [['a', 1], ['b', 3], ['a', 6], ['a', 2], ['b', 4], ['b', 5]]

from collections import defaultdict

dd = defaultdict(list)

for cat, num in first_list:
    dd[cat].append(num)

defaultdict(list, {'a': [1, 6, 2],
                   'b': [3, 4, 5]})

groupby

defaultdict解决方案具有O（ n ）复杂性，但是可以适当地命名itertools.groupby解决方案，这需要排序和O（ n log n ）复杂度：

from itertools import groupby
from operator import itemgetter

sorter = sorted(first_list, key=itemgetter(0))
grouper = groupby(sorter, key=itemgetter(0))
res = {i: list(map(itemgetter(1), j)) for i, j in grouper}

{'a': [1, 6, 2], 'b': [3, 4, 5]}

列表输出列表

与在list上调用dict.values一样简单：

res_list = list(res.values())

Answer 3

这是使用= Table.Group(#"Replaced Value", {"Company Name", "Policy Number", "Division", "Date"}, {{"Count", each List.NonNullCount([Claim Amount]), type number} })库和列表理解的绝佳机会：

itertools

我在这里所做的是按子列表中的第一个值对列表进行分组，然后提取第二个值。请注意，该列表需要预先排序，这样才能在first_list = [['a', 1], ['b', 3], ['a', 6], ['a', 2], ['b', 4], ['b', 5], ...] keyfunc = lambda x: x[0] new_list = [[v1[1] for v1 in v] for k, v in itertools.groupby(sorted(first_list, key = keyfunc), key = keyfunc)]时间运行。

Answer 4

reduce适用于任何数量的标签。

first_list = [['a', 1], ['b', 3], ['a', 6], ['a', 2], ['b', 4], ['b', 5]]
def lambda_group(acc, val):
    tag, x = val
    if key not in acc:
        acc[key] = []
    acc[key].append(value)
    return acc
grouped_vals = reduce(lambda_group, first_list, {})
regrouped = list(grouped_vals.values())

产生[[1, 6, 2], [3, 4, 5]]

Answer 5

好的，在python中有一个内置方法，但是以抽象算法的方式，我们可以说：

-mapView:rightCalloutAccessoryViewForAnnotation:

通过python

5 个答案:

defaultdict

groupby

列表输出列表