Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Chương trình Python sắp xếp ma trận theo tổng hàng


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_sum nhậ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.