python二维数组怎么按列排序

python

2597

2023/10/20 23:39:18

栏目: 编程语言

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

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

python二维数组怎么按列排序

最新知识库

相关标签