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

Python – Sắp xếp hàng ma trận theo số lượng phần tử tùy chỉnh


Sắp xếp các hàng của ma trận dựa trên số lượng phần tử xuất hiện trong một danh sách tùy chỉnh custom_list.

Ví dụ

def get_count_matrix(my_key):
    return len([element for element in my_key if element in custom_list])

my_list = [[31, 5, 22, 7], [85, 5], [9, 11, 22], [7, 48]]
print("Danh sách:", my_list)
custom_list = [31, 85, 7]
my_list.sort(key=get_count_matrix)
print("Kết quả:", my_list)

Kết quả

Danh sách: [[31, 5, 22, 7], [85, 5], [9, 11, 22], [7, 48]]
Kết quả: [[9, 11, 22], [85, 5], [7, 48], [31, 5, 22, 7]]

Giải thích

  • Hàm khóa đếm số phần tử của hàng nằm trong custom_list.

  • Sắp xếp tăng dần theo số lượng này.