我正在尝试使用自定义预定义列表作为所需顺序对元组的第一个元素上的元组列表进行排序。 所以我将列表作为
my_list=(['chr1',12],['chrX',32],['chr2',1],['chr1',79],['chr2',6])
并且第一个元素上的预定义列表是
custom_list=['chr1','chr2','chrX']
我希望输出为
(['chr1',12],['chr1',79],['chr2',1],['chr2',6],['chrX',32])
(目前,不需要对第二个元素进行排序。) 我无法弄清楚如何做到这一点。有人可以帮忙吗?
答案 0 :(得分:3)
您可以使用list.index()
功能将custom_list
中的位置转换为排序键:
sorted(my_list, key=lambda x: (custom_list.index(x[0]), x[1]))
您可能希望将custom_list
转换为字典,以便更快地进行映射:
custom_list_indices = {v: i for i, v in enumerate(custom_list)}
sorted(my_list, key=lambda x: (custom_list_indices.get(x[0]), x[1]))
字典查找需要不变的时间,list.index()
时间与列表的长度成正比。
另一个优点是,使用字典,您可以返回字典中未找到的条目的默认值(在此示例中为None
); list.index()
会引发ValueError
例外。
演示:
>>> my_list=(['chr1',12],['chrX',32],['chr2',1],['chr1',79],['chr2',6])
>>> custom_list=['chr1','chr2','chrX']
>>> sorted(my_list, key=lambda x: (custom_list.index(x[0]), x[1]))
[['chr1', 12], ['chr1', 79], ['chr2', 1], ['chr2', 6], ['chrX', 32]]
>>> custom_list_indices = {v: i for i, v in enumerate(custom_list)}
>>> sorted(my_list, key=lambda x: (custom_list_indices.get(x[0]), x[1]))
[['chr1', 12], ['chr1', 79], ['chr2', 1], ['chr2', 6], ['chrX', 32]]