python根据子列表值重新排列列表中的元素
Posted
技术标签:
【中文标题】python根据子列表值重新排列列表中的元素【英文标题】:python re-arrange elements in a list based on the sublist values 【发布时间】:2019-07-22 08:37:09 【问题描述】:我有一个列表列表,我想根据其子列表的值重新排列。
list = [["c","d"], ["b", "c"], ["a", "b"], ["x", "y"]]
最终产品应如下所示:
new_list = [["a", "b"], ["b", "c"], ["c", "d"], ["x", "y"]]
Python 代码应该分析list
中的每个元素,并将它们重新排列成new_list
。具有相同元素的子列表应并排放置,例如["b", "c"]
应放置在["a", "b"]
之后,形成["a", "b"], ["b", "c"]
的链。
这是我对列表进行排序的尝试:
for i in list:
if len(new_list) == 0:
new_list.append(i)
else:
for j in new_list:
if j[0] == i[1]:
newindex = new_list.index(j)
new_list.insert(newindex, i)
不幸的是,使用上面的代码,我得到了一个无限循环,它卡在 else 块中。
欢迎任何关于更好解决方案的建议。谢谢。
【问题讨论】:
一般来说,您不应该在迭代列表时修改列表的内容(插入/删除)。您还可以使用内置的sorted
和自定义按键功能。
不能只使用基本的 list.sort() 函数吗?我认为这会产生相同的结果?
您的排序顺序似乎未指定。如果你有子列表[ "a"," b"], [ "a"," c"], [ "b","c"]
?
new_list
应该是什么,例如list = [["b","m"], ["b","x"], ["b", "c"], ["a", "b"], ["x", "y"], ["m", "n"]]
? (也就是说 - 不止一条路径)。
这可能有点类似于Topological Sort,加深了你想用并行路径做什么
【参考方案1】:
基于@Adam.Er8 评论(我也有同样的想法,但他的例子比我的好):
>>> L = [["b","m"], ["b","x"], ["b", "c"], ["a", "b"], ["x", "y"], ["m", "n"]]
结果是一片森林。这个想法是从下到上构建树,使用 dict child -> parent
:
>>> d =
>>> for u, v in L:
... d[v] = u
... if not u in d.keys():
... d[u] = None
>>> d
'm': 'b', 'b': 'a', 'x': 'b', 'c': 'b', 'a': None, 'y': 'x', 'n': 'm'
现在,反转树很容易:
>>> e =
>>> for k, v in d.items():
... e.setdefault(v, []).append(k)
>>> e
'b': ['m', 'x', 'c'], 'a': ['b'], None: ['a'], 'x': ['y'], 'm': ['n']
使用 DFS,您可以输出路径:
>>> def dfs(x, stack=[]):
... if x in e:
... for y in e[x]:
... yield from dfs(y, stack + [x])
... else:
... yield stack + [x]
>>> P = list(dfs(None))
>>> P
[[None, 'a', 'b', 'm', 'n'], [None, 'a', 'b', 'x', 'y'], [None, 'a', 'b', 'c']]
最后:
>>> for p in P:
... print(list(zip(p[1:], p[2:])))
[('a', 'b'), ('b', 'm'), ('m', 'n')]
[('a', 'b'), ('b', 'x'), ('x', 'y')]
[('a', 'b'), ('b', 'c')]
【讨论】:
非常感谢您的回答。这很有帮助。我今天学到了一些新东西。以上是关于python根据子列表值重新排列列表中的元素的主要内容,如果未能解决你的问题,请参考以下文章
python中flat将嵌套列表中的元素按顺序排列在一个列表中