python二维数组怎么按列排序

avatar
作者
猴君
阅读量:1

在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,即按第一列进行排序。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!