Python: danh sách bộ lọc danh sách với một danh sách khác

Question

Tôi đang cố gắng lọc danh sách, tôi muốn trích xuất từ ​​danh sách A (danh sách danh sách), các phần tử khớp với chỉ mục chính 0, với danh sách B khác những gì có một serie các giá trị

như thế này

list_a = list(
    list(1, ...), 
    list(5, ...), 
    list(8, ...), 
    list(14, ...) 
) 

list_b = list(5, 8) 

return filter(lambda list_a: list_a[0] in list_b, list_a) 

nên quay lại:

list(
    list(5, ...), 
    list(8, ...) 
) 

làm thế nào tôi có thể làm điều này? Cảm ơn!

Answer 1

Sử dụng một danh sách hiểu:

result = [x for x in list_a if x[0] in list_b] 

Đối với cải thiện hiệu suất chuyển đổi list_b đến một tập đầu tiên.

Khi @kevin ghi chú trong các nhận xét giống như list(5,8) (trừ khi nó không phải là mã giả) không hợp lệ và bạn sẽ gặp lỗi.

list() chấp nhận chỉ có một mục và mục mà nên iterable/iterator

Answer 2

Bạn đang thực sự rất gần gũi. Chỉ cần làm điều này:

list_a = list(
    list(1, ...), 
    list(5, ...), 
    list(8, ...), 
    list(14, ...) 
) 

# Fix the syntax here 
list_b = [5, 8] 

return filter(lambda list_a: list_a[0] in list_b, list_a)