按列表对嵌套列表进行排序

Sort nested list by list

如何根据每个嵌套列表的第一个元素匹配顺序列表中给定的顺序对嵌套列表进行高效排序?

列表:[[97, 2, 0, 2], [97, 2, 0, 2], [98, 1, 2, 3], [99, 3, 3, 6], [99, 3, 3, 6], [99, 3, 3, 6], [101, 1, 6, 7], [100, 1, 7, 8]]

订单列表:[97, 98, 99, 99, 101, 100, 97, 99]

想要的列表:[[97, 2, 0, 2], [98, 1, 2, 3], [99, 3, 3, 6], [99, 3, 3, 6], [101, 1, 6, 7], [100, 1, 7, 8], [97, 2, 0, 2], [99, 3, 3, 6]]

尝试创建一个 dict 键控到嵌套列表中的第一个值。然后从 dict:

构建输出列表
nl = [[97, 2, 0, 2], [97, 2, 0, 2], [98, 1, 2, 3], [99, 3, 3, 6], [99, 3, 3, 6],
      [99, 3, 3, 6], [101, 1, 6, 7], [100, 1, 7, 8]]

# Associate first value in list to the list
d = {v[0]: v for v in nl}

order_lst = [97, 98, 99, 99, 101, 100, 97, 99]

# Grab the list associated to each value in order_list from d
out = [d[v] for v in order_lst]
print(out)

out:

[[97, 2, 0, 2], [98, 1, 2, 3], [99, 3, 3, 6], [99, 3, 3, 6], [101, 1, 6, 7], 
[100, 1, 7, 8], [97, 2, 0, 2], [99, 3, 3, 6]]

*请注意,这假定嵌套列表中的所有子列表都是可接受的值,因为 dict.

中不支持同一键的多个变体