在python中,可以使用sorted函數(shù)和lambda函數(shù)來按列對二維數(shù)組進(jìn)行排序。下面是一個示例代碼:
# 二維數(shù)組 matrix = [[5, 2, 3], [1, 7, 6], [4, 8, 9]] # 定義按列排序的函數(shù) def sort_by_column(arr, column): return sorted(arr, key=lambda x: x[column]) # 按第一列排序 sorted_matrix = sort_by_column(matrix, 0) print(sorted_matrix) # 輸出結(jié)果:[[1, 7, 6], [4, 8, 9], [5, 2, 3]]
登錄后復(fù)制
在上面的代碼中,我們定義了一個sort_by_column函數(shù),它接受一個二維數(shù)組和一個列索引作為參數(shù),然后使用sorted函數(shù)對二維數(shù)組進(jìn)行排序。lambda函數(shù)用于指定排序的key,即按照每個子數(shù)組的指定列進(jìn)行排序。最后,我們調(diào)用sort_by_column函數(shù),傳入二維數(shù)組和列索引0,即按第一列進(jìn)行排序。