2

我正在尝试使用自定义预定义列表作为所需顺序对元组的第一个元素上的元组列表进行排序。所以我的清单为

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])

（目前，不需要对第二个元素进行排序。）我不知道该怎么做。有人可以帮忙吗？

4

1 回答 1

5

您可以使用该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]]
于 2013-10-10T22:05:41.330 回答