使用现有列表打印新列表

Question

我想要一个这样的列表：['5', '0 1', '1 2', '1 8', '2 3'] 并返回一个由这样的元组组成的新列表： [(0, [1]),(1,[0,2,8]),(2[1,3]),(3,[2]),(8,[1])]。每个元组的第一个元素是一个整数，第二个元素是一个整数列表，它出现在原始列表中。我不能使用字典、集合、双端队列、二等分模块。

def create_network(file_name):

    friends = open(file_name).read().splitlines()

    network=[]

    for strings in friends:

        relationship=strings.strip().split(' ')

        if len(relationship)==2:

             a,b=relationship

             a=int(a)

             b=int(b)

             if a>=len(network):

                 network.append((a,[b]))

             else:

                 wow=network[a]

                 wow[1].append(b)

                 network[a]=wow

    return network

这是我到目前为止。我希望它返回： [(0, [1, 2, 3]), (1, [0, 4, 6, 7, 9]), (2, [0, 3, 6, 8, 9]) , (3, [0, 2, 8, 9]), (4, [1, 6, 7, 8]), (5, [9]), (6, [1, 2, 4, 8]) , (7, [1, 4, 8]), (8, [2, 3, 4, 6, 7]), (9, [1, 2, 3, 5])] 但它返回 [(0 , [1, 2, 3]), (1, [4, 6, 7, 9]), (2, [3, 6, 8, 9]), (3, [8, 9]), (4 , [6, 7, 8]), (5, [9]), (6, [8]), (7, [8])]。我不知道为什么它不起作用。

Answer 1

你可以这样做：from itertools import combinationsdef create_network(lst):    seen = {}    for e in lst:        for s, t in combinations(map(int, e.split()), 2):            seen.setdefault(s, set()).add(t)            seen.setdefault(t, set()).add(s)    return [(k, sorted(values)) for k, values in seen.items()]data = ['5', '0 1', '1 2', '1 8', '2 3']result = create_network(data)print(result)输出[(0, [1]), (1, [0, 2, 8]), (2, [1, 3]), (3, [2]), (8, [1])]一般的想法是创建一个字典，其中键是整数，值是它旁边出现的一组整数。该语句map(int, e.split())创建一个整数迭代，然后使用组合从迭代中挑选每一个可能的对，将每一对添加到字典中，最后返回值排序的元组。更新 （不使用任何内置模块）def combinations(m, lst):    if m == 0:        return [[]]    return [[x] + suffix for i, x in enumerate(lst) for suffix in combinations(m - 1, lst[i + 1:])]def create_network(lst):    uniques = []    for s in lst:        for e in map(int, s.split()):            if e not in uniques:                uniques.append(e)    result = []    for number in uniques:        seen = []        for e in lst:            values = list(map(int, e.split()))            for s, t in combinations(2, values):                if s == number:                    if t not in seen:                        seen.append(t)                elif t == number:                    if s not in seen:                        seen.append(s)        if seen:            result.append((number, sorted(seen)))    return sorted(result)data = ['5', '0 1', '1 2', '1 8', '2 3']network = create_network(data)print(network)输出[(0, [1]), (1, [0, 2, 8]), (2, [1, 3]), (3, [2]), (8, [1])]上面的代码没有使用 set、dictionary 和任何外部模块。请注意，它可能会很慢。

Answer 2

您可以使用列表理解：d = ['5', '0 1', '1 2', '1 8', '2 3']def find_edges(_d, c):  return [(a if b == c else b) for a, b in _d if c in [a, b]]new_d = [[int(c) for c in i.split()] for i in d if len(i) > 1]_final = []for i in [h for d in new_d for h in d]:   if not any(j == i for j, _ in _final):      _final.append((i, find_edges(new_d, i)))输出：[(0, [1]), (1, [0, 2, 8]), (2, [1, 3]), (8, [1]), (3, [2])]