python二维数组怎么按列排序

54次阅读

共计 438 个字符，预计需要花费 2 分钟才能阅读完成。

在 Python 中，可以使用 sorted 函数和 lambda 函数来按列对二维数组进行排序。下面是一个示例代码：

# 二维数组 
matrix = [[5, 2, 3],
          [1, 7, 6],
          [4, 8, 9]]

# 定义按列排序的函数 
def sort_by_column(arr, column):
    return sorted(arr, key=lambda x: x[column])

# 按第一列排序 
sorted_matrix = sort_by_column(matrix, 0)
print(sorted_matrix)

# 输出结果：[[1, 7, 6], [4, 8, 9], [5, 2, 3]]

在上面的代码中，我们定义了一个 sort_by_column 函数，它接受一个二维数组和一个列索引作为参数，然后使用 sorted 函数对二维数组进行排序。lambda 函数用于指定排序的 key，即按照每个子数组的指定列进行排序。最后，我们调用 sort_by_column 函数，传入二维数组和列索引 0，即按第一列进行排序。

丸趣 TV 网 – 提供最优质的资源集合！

正文完