溫馨提示×

python二維數(shù)組怎么按列排序

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

在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]]

在上面的代碼中，我們定義了一個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)行排序。

0 踩