根据第0个列表的argsort对嵌套列表中的列表进行排序

Question

根据列表中的第一个列表对列表嵌套进行排序的最巧妙的方法是什么？

排序之前

>>> list_of_things = [[100,300,200],[t,u,v]]

经过排序

>>> [[100,200,300],[t,v,u]]

到目前为止，我已经使用搜索找到了可以用来排序的索引

>>> import numpy as np
>>> sort_index=np.argsort(list_of_things[0]) 

但是我还没有找到一种在sort_index上为两个子列表建立索引的pythonic方法。

有什么想法吗？ 谢谢！

Answer 1

print(arr) 
array([['100', '300', '200'],
       ['t', 'u', 'v']],
      dtype='<U21')

首先， argsort第0 ^个列表进行argsort ：

idx = np.argsort(arr[0])

print(idx)
array([0, 2, 1])

现在，只需使用numpy indexing ：

arr = arr[:, idx]

print(arr)
array([['100', '200', '300'],
       ['t', 'v', 'u']],
      dtype='<U21')

Answer 2

尝试使用来自more_itertools的工具，该工具是more_itertools可迭代对象进行并行排序的第三方库。

首先安装库：

> pip install more_itertools

码

from more_itertools import sort_together


iterables = [[100, 300, 200], ["t", "u", "v"]]
sort_together(iterables)
# [(100, 200, 300), ('t', 'v', 'u')]

默认情况下，所有可迭代项均按第0个索引排序。 可以通过key_list参数调整此排序优先级。

有关more_itertools信息，请参见more_itertools文档 。