按第一个列表对列表进行排序

Question

我目前编写了一个小脚本，其中包含四个列表，例如，每个第 0 个索引相互匹配。我按第一个列表对这些列表进行排序，而其他列表以相同的模式排序。像这样：

未排序的列表：

list0: 4, 2, 1, 3

list1：四、二、一、三

排序列表：

list0: 1, 2, 3, 4

list1：一、二、三、四

这适用于以下脚本：

list0=IN[0]
list1=IN[1]
list2=IN[2]
list3=IN[3]

list0,list1,list2,list3 =zip(*sorted(zip(list0,list1,list2,list3)))

list0,list1,list2,list3 = (list(t) for t in zip(*sorted(zip(list0,list1,list2,list3))))


#Output:
OUT = [list0,list1,list2,list3]

我希望能够将其转换为具有任意数量列表的更通用的脚本。因此，我的输入将是一个列表列表，它们应该按照上面的代码所示进行排序。你能帮我吗？

最好的问候，

问

Answer 1

这是你要找的吗？

def sort_lists(*args):
    zipped_list = zip(*sorted(zip(*args)))
    return [list(l) for l in zipped_list]

例子：

>>> l1 = [4,2,1,3]
>>> l2 = ["four", "two", "one", "three"]
>>> sort_lists(l1, l2)
[[1, 2, 3, 4], ['one', 'two', 'three', 'four']]

根据您的代码示例，您可以按如下方式使用它：

>>> OUT = sort_lists(*IN)

Answer 2

问题的另一个答案：

list_of_lists = [
    [1, 2, 3, 0],
    [1, 2, 3, 4],
    [5, 6, 7, 8]]

dict_of_lists = {
    'a': [1, 2, 3, 0],
    'b': [1, 2, 3, 4],
    'c': [5, 6, 7, 8]}


def sort_list_of_lists(lst, idx):
    # idx - is index of list to use for sorting all the lists
    return list(list(tup) for tup in zip(*sorted(zip(*lst), key=lambda x: x[idx])))


def sort_dict_of_lists(dct, key):
    # key - is key of list to use for sorting all the lists
    keys = list(dct.keys())
    # get list from dict
    lst = [dct[key_] for key_ in keys]
    # get idx from key
    idx = keys.index(key)
    # use function 'sort_list_of_lists' for actual sorting
    return {key_: lst_ for key_, lst_ in zip(keys, sort_list_of_lists(lst, idx))}


sorted_list_of_lists = sort_list_of_lists(list_of_lists, idx=0)
sorted_dict_of_lists = sort_dict_of_lists(dict_of_lists, key='a')

如果我遇到这样的问题，我会使用pandas 来解决它 - 处理表格的非常方便高效的工具：

import pandas as pd

dataframe_from_list = pd.DataFrame(list_of_lists).T
dataframe_from_dict = pd.DataFrame(dict_of_lists)

dataframe_from_list.sort_values(by=0, inplace=True)
dataframe_from_dict.sort_values(by='a', inplace=True)