python如何给二维列表排序

使用Python进行二维列表排序的方法有很多种，包括使用内置函数sorted、使用lambda函数、以及自定义排序函数等。最常见的方式是使用sorted函数结合lambda函数来指定排序的关键字。其中，使用lambda函数是最常用且简单的方法。例如，可以根据列表的某一列进行排序，或者根据多列进行排序。

以下是使用Python对二维列表进行排序的详细介绍：

一、使用sorted函数和lambda函数进行排序

使用Python的内置函数sorted结合lambda函数是最常见的方式。lambda函数可以用来指定排序的关键字，这使得排序变得非常灵活。

1、按单列排序

假设我们有一个二维列表，其中每个子列表包含三个元素。我们希望根据第二个元素对列表进行排序。

data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6]
]
按第二个元素排序
sorted_data = sorted(data, key=lambda x: x[1])
print(sorted_data)

在上面的代码中，key=lambda x: x[1]表示我们根据每个子列表的第二个元素进行排序。

2、按多列排序

有时候，我们可能需要根据多列进行排序。例如，先根据第二列排序，如果第二列相同，再根据第三列排序。

data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6],
    [2, 3, 5]
]
按第二列和第三列排序
sorted_data = sorted(data, key=lambda x: (x[1], x[2]))
print(sorted_data)

在这个例子中，key=lambda x: (x[1], x[2])表示我们先根据第二个元素排序，如果第二个元素相同，再根据第三个元素排序。

二、使用内置sort方法进行排序

除了使用sorted函数之外，我们还可以使用列表的内置sort方法进行排序。与sorted函数不同，sort方法会对原列表进行排序，不会返回一个新的列表。

1、按单列排序

data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6]
]
按第二个元素排序
data.sort(key=lambda x: x[1])
print(data)

2、按多列排序

data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6],
    [2, 3, 5]
]
按第二列和第三列排序
data.sort(key=lambda x: (x[1], x[2]))
print(data)

三、自定义排序函数

有时候，排序规则比较复杂，无法简单地用lambda函数表达。这时，我们可以定义一个自定义的排序函数，然后传递给sorted或sort方法。

1、按单列排序

def custom_sort(row):
    return row[1]
data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6]
]
使用自定义排序函数
sorted_data = sorted(data, key=custom_sort)
print(sorted_data)

2、按多列排序

def custom_sort(row):
    return (row[1], row[2])
data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6],
    [2, 3, 5]
]
使用自定义排序函数
sorted_data = sorted(data, key=custom_sort)
print(sorted_data)

四、倒序排序

有时我们需要对列表进行降序排序，可以在sorted或sort方法中传递reverse=True参数。

1、按单列降序排序

data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6]
]
按第二个元素降序排序
sorted_data = sorted(data, key=lambda x: x[1], reverse=True)
print(sorted_data)

2、按多列降序排序

data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6],
    [2, 3, 5]
]
按第二列和第三列降序排序
sorted_data = sorted(data, key=lambda x: (x[1], x[2]), reverse=True)
print(sorted_data)

五、结合条件进行排序

有时候，我们需要根据某些条件对列表进行排序，例如只对满足某个条件的行进行排序，其他行保持原样。

data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6],
    [2, 3, 5]
]
只对第二列大于4的行进行排序
sorted_data = sorted(data, key=lambda x: x[1] if x[1] > 4 else float('inf'))
print(sorted_data)

在这个例子中，x[1] if x[1] > 4 else float('inf')表示只有当第二列大于4时，才对该行进行排序，否则保持原样。

六、使用pandas进行排序

对于更复杂的二维数据处理，使用pandas库可能更为方便。pandas提供了强大的数据操作功能，包括排序。

1、按单列排序

import pandas as pd
data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6]
]
df = pd.DataFrame(data, columns=['A', 'B', 'C'])
sorted_df = df.sort_values(by='B')
print(sorted_df)

2、按多列排序

import pandas as pd
data = [
    [1, 5, 9],
    [2, 3, 8],
    [4, 7, 6],
    [2, 3, 5]
]
df = pd.DataFrame(data, columns=['A', 'B', 'C'])
sorted_df = df.sort_values(by=['B', 'C'])
print(sorted_df)