Khi cần sắp xếp ma trận (danh sách lồng nhau) dựa trên tổng của mỗi hàng, ta định nghĩa một hàm sử dụng phương thức sum() làm khóa sắp xếp.
Ví dụ
def sort_sum(row):
return sum(row)
my_list = [[34, 51], [32, 15, 67], [12, 41], [54, 36, 22]]
print("Danh sách ban đầu:")
print(my_list)
my_list.sort(key=sort_sum)
print("Kết quả:")
print(my_list)
Kết quả
Danh sách ban đầu: [[34, 51], [32, 15, 67], [12, 41], [54, 36, 22]] Kết quả: [[12, 41], [34, 51], [54, 36, 22], [32, 15, 67]]
Giải thích
Định nghĩa hàm
sort_sumnhận một hàng (list) và trả về tổng các phần tử.Định nghĩa ma trận (danh sách các list) và hiển thị.
Gọi
sort()với tham sốkey=sort_sumđể sắp xếp theo tổng hàng.In ma trận đã sắp xếp.