溫馨提示×

python二維數組怎么按列排序

小億
189
2023-10-31 23:39:18
欄目: 編程語言

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

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女