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