首頁 >後端開發 >Python教學 >python二維數組怎麼按列排序

python二維數組怎麼按列排序

WBOY
WBOY轉載
2024-03-01 14:49:30579瀏覽

python二維數組怎麼按列排序

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,也就是按第一列進行排序。

以上是python二維數組怎麼按列排序的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:lsjlt.com。如有侵權,請聯絡admin@php.cn刪除