python 二维数组排序

python 二维数组排序

在python中，对二维数组进行排序是一个常见的操作。二维数组也被称为矩阵，是一个由多行和多列组成的数组。在本文中，我们将讨论如何对二维数组进行排序，并给出一些实际的示例。

1. 对二维数组按行或列进行排序

在python中，我们可以使用内置的sorted()函数来对二维数组进行排序。当我们想按行或列对二维数组进行排序时，我们可以使用key参数来指定排序的规则。

1.1 对二维数组按行进行排序

首先，让我们看一个简单的示例，如何对二维数组按行进行排序。假设我们有一个包含多个列表的二维数组，我们希望按照每行第一个元素的大小对数组进行排序。

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

sorted_matrix = sorted(matrix, key=lambda x: x[0])
print(sorted_matrix)

运行以上代码，我们会得到按照每行第一个元素的大小对二维数组进行排序后的结果：

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

在这个示例中，我们使用了lambda函数作为key参数，它代表每一行元素的第一个元素。通过这种方式，我们实现了按行进行排序。

1.2 对二维数组按列进行排序

除了按行进行排序，我们也可以按列进行排序。下面是一个示例，如何对二维数组按照每列的第一个元素进行排序。

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

sorted_matrix = sorted(matrix, key=lambda x: x[0])
sorted_matrix_transposed = [list(i) for i in zip(*sorted_matrix)]
print(sorted_matrix_transposed)

运行以上代码，我们会得到按照每列第一个元素的大小对二维数组进行排序后的结果：

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

在这个示例中，首先我们对二维数组按行进行排序，然后使用zip(*sorted_matrix)将排好序的二维数组转置，最终得到按列排序的结果。

2. 使用numpy库对二维数组进行排序

除了使用内置的sorted()函数，我们还可以使用numpy库中的函数对二维数组进行排序。numpy是python中一个强大的数学库，提供了大量的数学函数和操作。

下面是一个使用numpy库对二维数组进行排序的示例：

import numpy as np

matrix = np.array([[3, 2, 5],
                    [1, 7, 4],
                    [6, 8, 0]])

sorted_matrix = np.sort(matrix, axis=0)
print(sorted_matrix)

运行以上代码，我们会得到按列对二维数组进行排序后的结果：

[[1 2 0]
 [3 7 4]
 [6 8 5]]

在这个示例中，我们使用np.sort()函数对二维数组按列进行排序。axis=0参数表示按列排序，axis=1表示按行排序。

3. 总结

在本文中，我们详细讨论了如何使用python对二维数组进行排序。我们介绍了如何按行或列进行排序，并给出了使用内置函数和numpy库的具体示例。