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






