共計(jì) 438 個(gè)字符,預(yù)計(jì)需要花費(fèi) 2 分鐘才能閱讀完成。
在 Python 中,可以使用 sorted 函數(shù)和 lambda 函數(shù)來(lái)按列對(duì)二維數(shù)組進(jìn)行排序。下面是一個(gè)示例代碼:
# 二維數(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]]
在上面的代碼中,我們定義了一個(gè) sort_by_column 函數(shù),它接受一個(gè)二維數(shù)組和一個(gè)列索引作為參數(shù),然后使用 sorted 函數(shù)對(duì)二維數(shù)組進(jìn)行排序。lambda 函數(shù)用于指定排序的 key,即按照每個(gè)子數(shù)組的指定列進(jìn)行排序。最后,我們調(diào)用 sort_by_column 函數(shù),傳入二維數(shù)組和列索引 0,即按第一列進(jìn)行排序。
丸趣 TV 網(wǎng) – 提供最優(yōu)質(zhì)的資源集合!
正文完