python二维数组怎么按列排序

39次阅读
没有评论

共计 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 网 – 提供最优质的资源集合!

正文完
 
丸趣
版权声明:本站原创文章,由 丸趣 2023-12-16发表,共计438字。
转载说明:除特殊说明外本站除技术相关以外文章皆由网络搜集发布,转载请注明出处。
评论(没有评论)