Python – Lọc hàng chứa các phần tử bắt buộc

Khi cần lọc các hàng mà chứa tất cả các phần tử bắt buộc từ một danh sách kiểm tra, ta dùng list comprehension và toán tử all().

Ví dụ

my_list = [[261, 49, 61], [27, 49, 3, 261], [261, 49, 85], [1, 1, 9]]
print("Danh sách ban đầu:")
print(my_list)
check_list = [49, 61, 261, 85]
my_result = [index for index in my_list if all(element in check_list for element in index)]
print("Kết quả:")
print(my_result)

Kết quả

Danh sách ban đầu: [[261, 49, 61], [27, 49, 3, 261], [261, 49, 85], [1, 1, 9]]
Kết quả: [[261, 49, 61], [261, 49, 85]]

Giải thích

• Định nghĩa danh sách lồng nhau và danh sách kiểm tra.

• Duyệt từng hàng, kiểm tra all(element in check_list for element in row).

• Chỉ giữ hàng mà mọi phần tử đều nằm trong check_list.

• In kết quả.