Làm thế nào để kiểm tra nếu một trong các mục sau đây có trong danh sách không?

Tôi đang cố gắng tìm một cách ngắn để xem liệu có bất kỳ mục nào sau đây có trong danh sách không, nhưng lần thử đầu tiên của tôi không hoạt động. Bên cạnh việc viết một hàm để thực hiện điều này, là bất kỳ cách ngắn nào để kiểm tra xem một trong nhiều mục có trong danh sách hay không.

>>> a = [2,3,4]
>>> print (1 or 2) in a
False
>>> print (2 or 1) in a
True

>>> L1 = [2,3,4]
>>> L2 = [1,2]
>>> [i for i in L1 if i in L2]
[2]


>>> S1 = set(L1)
>>> S2 = set(L2)
>>> S1.intersection(S2)
set([2])

Cả danh sách trống và bộ trống đều là Sai, vì vậy bạn có thể sử dụng giá trị trực tiếp làm giá trị thật.

Ah, Tobias bạn đánh tôi với nó. Tôi đã nghĩ về sự thay đổi nhỏ này trên giải pháp của bạn:

>>> a = [1,2,3,4]
>>> b = [2,7]
>>> print(any(x in a for x in b))
True

Có lẽ lười hơn một chút:

a = [1,2,3,4]
b = [2,7]

print any((True for x in a if x in b))

Hãy suy nghĩ về những gì mã thực sự nói!

>>> (1 or 2)
1
>>> (2 or 1)
2

Điều đó có lẽ nên giải thích nó. :) Python rõ ràng thực hiện "lười hoặc", điều này sẽ không gây ngạc nhiên. Nó thực hiện nó một cái gì đó như thế này:

def or(x, y):
    if x: return x
    if y: return y
    return False

Trong ví dụ đầu tiên, x == 1và y == 2. Trong ví dụ thứ hai, nó ngược lại. Đó là lý do tại sao nó trả về các giá trị khác nhau tùy thuộc vào thứ tự của chúng.

a = {2,3,4}
if {1,2} & a:
    pass

Mã phiên bản golf. Cân nhắc sử dụng một bộ nếu nó có ý nghĩa để làm như vậy. Tôi thấy điều này dễ đọc hơn một sự hiểu biết danh sách.

1 dòng không có danh sách hiểu.

>>> any(map(lambda each: each in [2,3,4], [1,2]))
True
>>> any(map(lambda each: each in [2,3,4], [1,5]))
False
>>> any(map(lambda each: each in [2,3,4], [2,4]))
True

Tốt nhất tôi có thể đến với:

any([True for e in (1, 2) if e in a])

Trong python 3, chúng ta có thể bắt đầu sử dụng dấu hoa thị. Đưa ra hai danh sách:

bool(len({*a} & {*b}))

Chỉnh sửa: kết hợp đề xuất của alkanen

Khi bạn nghĩ "kiểm tra xem nếu a trong b", hãy nghĩ băm (trong trường hợp này là bộ). Cách nhanh nhất là băm danh sách bạn muốn kiểm tra, sau đó kiểm tra từng mục trong đó.

Đây là lý do tại sao câu trả lời của Joe Koberg rất nhanh: kiểm tra giao lộ tập hợp rất nhanh.

Khi bạn không có nhiều dữ liệu, việc tạo các bộ có thể gây lãng phí thời gian. Vì vậy, bạn có thể tạo một bộ danh sách và chỉ cần kiểm tra từng mục:

tocheck = [1,2] # items to check
a = [2,3,4] # the list

a = set(a) # convert to set (O(len(a)))
print [i for i in tocheck if i in a] # check items (O(len(tocheck)))

Khi số lượng mặt hàng bạn muốn kiểm tra là nhỏ, sự khác biệt có thể không đáng kể. Nhưng hãy kiểm tra nhiều số với một danh sách lớn ...

kiểm tra:

from timeit import timeit

methods = ['''tocheck = [1,2] # items to check
a = [2,3,4] # the list
a = set(a) # convert to set (O(n))
[i for i in tocheck if i in a] # check items (O(m))''',

'''L1 = [2,3,4]
L2 = [1,2]
[i for i in L1 if i in L2]''',

'''S1 = set([2,3,4])
S2 = set([1,2])
S1.intersection(S2)''',

'''a = [1,2]
b = [2,3,4]
any(x in a for x in b)''']

for method in methods:
    print timeit(method, number=10000)

print

methods = ['''tocheck = range(200,300) # items to check
a = range(2, 10000) # the list
a = set(a) # convert to set (O(n))
[i for i in tocheck if i in a] # check items (O(m))''',

'''L1 = range(2, 10000)
L2 = range(200,300)
[i for i in L1 if i in L2]''',

'''S1 = set(range(2, 10000))
S2 = set(range(200,300))
S1.intersection(S2)''',

'''a = range(200,300)
b = range(2, 10000)
any(x in a for x in b)''']

for method in methods:
    print timeit(method, number=1000)

tốc độ:

M1: 0.0170331001282 # make one set
M2: 0.0164539813995 # list comprehension
M3: 0.0286040306091 # set intersection
M4: 0.0305438041687 # any

M1: 0.49850320816 # make one set
M2: 25.2735087872 # list comprehension
M3: 0.466138124466 # set intersection
M4: 0.668627977371 # any

Phương pháp luôn nhanh là tạo một bộ (trong danh sách), nhưng giao điểm hoạt động trên các tập dữ liệu lớn là tốt nhất!

Trong một số trường hợp (ví dụ: các thành phần danh sách duy nhất), có thể sử dụng các thao tác thiết lập.

>>> a=[2,3,4]
>>> set(a) - set([2,3]) != set(a)
True
>>> 

Hoặc, bằng cách sử dụng set.itorisjoint () ,

>>> not set(a).isdisjoint(set([2,3]))
True
>>> not set(a).isdisjoint(set([5,6]))
False
>>> 

Điều này sẽ làm điều đó trong một dòng.

>>> a=[2,3,4]
>>> b=[1,2]
>>> bool(sum(map(lambda x: x in b, a)))
True

Tôi đã thu thập một số giải pháp được đề cập trong các câu trả lời khác và trong các bình luận, sau đó chạy thử nghiệm tốc độ. not set(a).isdisjoint(b)Hóa ra là nhanh nhất, nó cũng không chậm lại nhiều khi kết quả làFalse .

Mỗi trong ba lần chạy kiểm tra một mẫu nhỏ về các cấu hình có thể có avà b. Thời gian tính bằng micrô giây.

Any with generator and max
        2.093 1.997 7.879
Any with generator
        0.907 0.692 2.337
Any with list
        1.294 1.452 2.137
True in list
        1.219 1.348 2.148
Set with &
        1.364 1.749 1.412
Set intersection explcit set(b)
        1.424 1.787 1.517
Set intersection implicit set(b)
        0.964 1.298 0.976
Set isdisjoint explicit set(b)
        1.062 1.094 1.241
Set isdisjoint implicit set(b)
        0.622 0.621 0.753

import timeit

def printtimes(t):
    print '{:.3f}'.format(t/10.0),

setup1 = 'a = range(10); b = range(9,15)'
setup2 = 'a = range(10); b = range(10)'
setup3 = 'a = range(10); b = range(10,20)'

print 'Any with generator and max\n\t',
printtimes(timeit.Timer('any(x in max(a,b,key=len) for x in min(b,a,key=len))',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('any(x in max(a,b,key=len) for x in min(b,a,key=len))',setup=setup2).timeit(10000000))
printtimes(timeit.Timer('any(x in max(a,b,key=len) for x in min(b,a,key=len))',setup=setup3).timeit(10000000))
print

print 'Any with generator\n\t',
printtimes(timeit.Timer('any(i in a for i in b)',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('any(i in a for i in b)',setup=setup2).timeit(10000000))
printtimes(timeit.Timer('any(i in a for i in b)',setup=setup3).timeit(10000000))
print

print 'Any with list\n\t',
printtimes(timeit.Timer('any([i in a for i in b])',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('any([i in a for i in b])',setup=setup2).timeit(10000000))
printtimes(timeit.Timer('any([i in a for i in b])',setup=setup3).timeit(10000000))
print

print 'True in list\n\t',
printtimes(timeit.Timer('True in [i in a for i in b]',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('True in [i in a for i in b]',setup=setup2).timeit(10000000))
printtimes(timeit.Timer('True in [i in a for i in b]',setup=setup3).timeit(10000000))
print

print 'Set with &\n\t',
printtimes(timeit.Timer('bool(set(a) & set(b))',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('bool(set(a) & set(b))',setup=setup2).timeit(10000000))
printtimes(timeit.Timer('bool(set(a) & set(b))',setup=setup3).timeit(10000000))
print

print 'Set intersection explcit set(b)\n\t',
printtimes(timeit.Timer('bool(set(a).intersection(set(b)))',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('bool(set(a).intersection(set(b)))',setup=setup2).timeit(10000000))
printtimes(timeit.Timer('bool(set(a).intersection(set(b)))',setup=setup3).timeit(10000000))
print

print 'Set intersection implicit set(b)\n\t',
printtimes(timeit.Timer('bool(set(a).intersection(b))',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('bool(set(a).intersection(b))',setup=setup2).timeit(10000000))
printtimes(timeit.Timer('bool(set(a).intersection(b))',setup=setup3).timeit(10000000))
print

print 'Set isdisjoint explicit set(b)\n\t',
printtimes(timeit.Timer('not set(a).isdisjoint(set(b))',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('not set(a).isdisjoint(set(b))',setup=setup2).timeit(10000000))
printtimes(timeit.Timer('not set(a).isdisjoint(set(b))',setup=setup3).timeit(10000000))
print

print 'Set isdisjoint implicit set(b)\n\t',
printtimes(timeit.Timer('not set(a).isdisjoint(b)',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('not set(a).isdisjoint(b)',setup=setup1).timeit(10000000))
printtimes(timeit.Timer('not set(a).isdisjoint(b)',setup=setup3).timeit(10000000))
print

Tôi phải nói rằng tình huống của tôi có thể không phải là điều bạn đang tìm kiếm, nhưng nó có thể cung cấp một sự thay thế cho suy nghĩ của bạn.

Tôi đã thử cả phương thức set () và any () nhưng vẫn gặp vấn đề với tốc độ. Vì vậy, tôi nhớ Raymond Hettinger nói rằng mọi thứ trong python đều là từ điển và sử dụng dict bất cứ khi nào bạn có thể. Vì vậy, đó là những gì tôi đã cố gắng.

Tôi đã sử dụng một defaultdict với int để chỉ ra kết quả âm tính và sử dụng mục trong danh sách đầu tiên làm khóa cho danh sách thứ hai (được chuyển đổi thành defaultdict). Bởi vì bạn đã tra cứu ngay lập tức với dict, bạn biết ngay rằng mục đó có tồn tại trong defaultdict hay không. Tôi biết bạn không phải lúc nào cũng thay đổi cấu trúc dữ liệu cho danh sách thứ hai của mình, nhưng nếu bạn có thể ngay từ đầu, thì nó sẽ nhanh hơn nhiều. Bạn có thể phải chuyển đổi list2 (danh sách lớn hơn) thành defaultdict, trong đó khóa là giá trị tiềm năng bạn muốn kiểm tra từ danh sách nhỏ và giá trị là 1 (nhấn) hoặc 0 (không nhấn, mặc định).

from collections import defaultdict
already_indexed = defaultdict(int)

def check_exist(small_list, default_list):
    for item in small_list:
        if default_list[item] == 1:
            return True
    return False

if check_exist(small_list, already_indexed):
    continue
else:
    for x in small_list:
        already_indexed[x] = 1

Đơn giản.

_new_list = []
for item in a:
    if item in b:
        _new_list.append(item)
    else:
        pass