Làm thế nào để kiểm tra nếu tất cả các yếu tố của một danh sách phù hợp với một điều kiện?

Tôi có một danh sách bao gồm khoảng 20000 danh sách. Tôi sử dụng phần tử thứ 3 của mỗi danh sách làm cờ. Tôi muốn thực hiện một số thao tác trong danh sách này miễn là ít nhất một cờ của phần tử là 0, giống như:

my_list = [["a", "b", 0], ["c", "d", 0], ["e", "f", 0], .....]

Ban đầu, tất cả các cờ đều bằng 0. Tôi sử dụng vòng lặp while để kiểm tra xem ít nhất một cờ của phần tử có bằng 0 hay không:

def check(list_):
    for item in list_:
        if item[2] == 0:
            return True
    return False

Nếu check(my_list)trả về True, sau đó tôi tiếp tục làm việc trong danh sách của mình:

while check(my_list):
    for item in my_list:
        if condition:
            item[2] = 1
        else:
            do_sth()

Trên thực tế, tôi muốn xóa một phần tử trong my_list khi tôi lặp lại phần tử đó, nhưng tôi không được phép xóa các mục khi tôi lặp lại phần tử đó.

Bản gốc my_list không có cờ:

my_list = [["a", "b"], ["c", "d"], ["e", "f"], .....]

Vì tôi không thể loại bỏ các yếu tố khi tôi lặp lại nó, tôi đã phát minh ra những lá cờ này. Nhưng trong đó my_listcó nhiều mục và whilevòng lặp đọc tất cả chúng ở mỗi forvòng lặp và nó tiêu tốn rất nhiều thời gian! Bạn có đề nghị nào không?

Câu trả lời tốt nhất ở đây là sử dụng all(), đó là nội dung cho tình huống này. Chúng tôi kết hợp điều này với một biểu thức trình tạo để tạo ra kết quả bạn muốn sạch sẽ và hiệu quả. Ví dụ:

>>> items = [[1, 2, 0], [1, 2, 0], [1, 2, 0]]
>>> all(flag == 0 for (_, _, flag) in items)
True
>>> items = [[1, 2, 0], [1, 2, 1], [1, 2, 0]]
>>> all(flag == 0 for (_, _, flag) in items)
False

Lưu ý rằng all(flag == 0 for (_, _, flag) in items)tương đương trực tiếp với all(item[2] == 0 for item in items), nó chỉ đẹp hơn một chút để đọc trong trường hợp này.

Và, đối với ví dụ về bộ lọc, việc hiểu danh sách (tất nhiên, bạn có thể sử dụng biểu thức trình tạo khi thích hợp):

>>> [x for x in items if x[2] == 0]
[[1, 2, 0], [1, 2, 0]]

Nếu bạn muốn kiểm tra ít nhất một phần tử là 0, tùy chọn tốt hơn là sử dụng phần tử any()nào dễ đọc hơn:

>>> any(flag == 0 for (_, _, flag) in items)
True

Nếu bạn muốn kiểm tra xem có mục nào trong danh sách vi phạm điều kiện sử dụng không all:

if all([x[2] == 0 for x in lista]):
    # Will run if all elements in the list has x[2] = 0 (use not to invert if necessary)

Để loại bỏ tất cả các yếu tố không phù hợp, sử dụng filter

# Will remove all elements where x[2] is 0
listb = filter(lambda x: x[2] != 0, listb)

Bạn có thể sử dụng itertools như thế này, nó sẽ dừng lại khi một điều kiện được đáp ứng mà không tuyên bố của bạn. Phương pháp ngược lại sẽ là thả

for x in itertools.takewhile(lambda x: x[2] == 0, list)
    print x

Một cách khác để sử dụng itertools.ifilter. Điều này kiểm tra tính trung thực và quy trình (sử dụng lambda)

Mẫu vật-

for x in itertools.ifilter(lambda x: x[2] == 0, my_list):
    print x

cách này linh hoạt hơn một chút so với sử dụng all():

my_list = [[1, 2, 0], [1, 2, 0], [1, 2, 0]]
all_zeros = False if False in [x[2] == 0 for x in my_list] else True
any_zeros = True if True in [x[2] == 0 for x in my_list] else False

hoặc ngắn gọn hơn:

all_zeros = not False in [x[2] == 0 for x in my_list]
any_zeros = 0 in [x[2] for x in my_list]