如何让listTwo跟随listOne的重排规则同步调整顺序,无需复杂操作
同步重排两个列表的极简实现
以下方案均使用Python内置函数实现,无需额外引入依赖,也不需要复杂的循环判断逻辑。
场景1:listOne元素无重复(符合你给出的示例场景)
先构建listOne元素到listTwo对应位置元素的映射,处理完listOne后直接拉取映射值生成新的listTwo,核心仅需2行代码:
# 调用MyFunction前先构建元素映射 elem_map = dict(zip(listOne, listTwo)) # 调用MyFunction得到重排后的listOne后,直接生成同步重排的listTwo listTwo = [elem_map[item] for item in listOne]
你的示例运行后,得到的listTwo就是[['e'], ['d'], ['f']],完全符合预期,时间复杂度为O(n),效率很高。
场景2:listOne存在重复元素
如果listOne有重复值无法作为字典的键,就通过索引同步重排:
# 打包原索引与listOne元素 indexed_list = list(enumerate(listOne)) # 调用MyFunction对indexed_list按原规则重排(只需将原逻辑中对元素的判断改为对元组第二位的判断即可) reordered_indexed = MyFunction(indexed_list) # 拆分得到重排后的索引序列,直接索引listTwo得到结果 reordered_idx = [x[0] for x in reordered_indexed] listOne = [x[1] for x in reordered_indexed] listTwo = [listTwo[i] for i in reordered_idx]
极简一次性方案
如果可以调整MyFunction的入参逻辑,直接打包两个列表同步重排即可:
paired = list(zip(listOne, listTwo)) # 修改MyFunction的重排逻辑,针对每个元组的第一位(原listOne元素)排序 reordered_paired = MyFunction(paired) listOne, listTwo = map(list, zip(*reordered_paired))
内容的提问来源于stack exchange,提问作者SeekNDstroy
相关产品推荐
相关产品推荐

