จะตรวจสอบว่าหนึ่งในรายการต่อไปนี้อยู่ในรายการได้อย่างไร?

ฉันกำลังพยายามหาวิธีสั้น ๆ เพื่อดูว่ามีรายการใดต่อไปนี้อยู่ในรายการหรือไม่ แต่ความพยายามครั้งแรกของฉันไม่ทำงาน นอกเหนือจากการเขียนฟังก์ชั่นเพื่อทำให้สิ่งนี้สำเร็จแล้วเป็นวิธีที่สั้น ๆ ในการตรวจสอบว่าหนึ่งในหลาย ๆ รายการอยู่ในรายการหรือไม่

>>> 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])

ทั้งรายการว่างและชุดว่างเป็นเท็จดังนั้นคุณสามารถใช้ค่าโดยตรงเป็นค่าความจริง

โทเบียสคุณเอาชนะฉันให้ได้ ฉันกำลังคิดถึงการเปลี่ยนแปลงเล็กน้อยในโซลูชันของคุณ:

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

อาจจะขี้เกียจนิดหน่อย:

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

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

ลองนึกถึงสิ่งที่รหัสพูดจริง ๆ !

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

นั่นอาจจะอธิบายได้ :) Python ใช้ "ขี้เกียจหรือ" อย่างเห็นได้ชัดซึ่งไม่น่าแปลกใจเลย มันทำอะไรแบบนี้:

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

ในตัวอย่างแรกและx == 1 y == 2ในตัวอย่างที่สองมันกลับกัน นั่นเป็นเหตุผลที่มันส่งกลับค่าที่แตกต่างกันขึ้นอยู่กับลำดับของพวกเขา

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

รุ่นรหัสกอล์ฟ พิจารณาใช้ชุดถ้ามันเหมาะสมที่จะทำ ฉันพบว่าสิ่งนี้สามารถอ่านได้มากกว่าความเข้าใจในรายการ

1 บรรทัดโดยไม่มีความเข้าใจในรายการ

>>> 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

ดีที่สุดที่ฉันจะได้มากับ:

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

ใน python 3 เราสามารถเริ่มต้นใช้เครื่องหมายดอกจันได้ รับสองรายการ:

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

แก้ไข: รวมข้อเสนอแนะของ alkanen

เมื่อคุณคิดว่า "ตรวจสอบเพื่อดูว่า a ในข" ให้คิดแฮช (ในกรณีนี้คือชุด) วิธีที่เร็วที่สุดคือการแฮชรายการที่คุณต้องการตรวจสอบแล้วตรวจสอบแต่ละรายการในนั้น

นี่คือสาเหตุที่คำตอบของ Joe Koberg รวดเร็ว: การตรวจสอบจุดตัดชุดนั้นเร็วมาก

เมื่อคุณไม่มีข้อมูลจำนวนมากการจัดฉากอาจทำให้เสียเวลา ดังนั้นคุณสามารถสร้างรายการและตรวจสอบแต่ละรายการ:

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)))

เมื่อจำนวนรายการที่คุณต้องการตรวจสอบมีขนาดเล็กความแตกต่างอาจเล็กน้อย แต่ตรวจสอบตัวเลขจำนวนมากกับรายการที่มีขนาดใหญ่ ...

การทดสอบ:

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)

ความเร็ว:

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

วิธีการที่รวดเร็วอย่างต่อเนื่องคือการสร้างหนึ่งชุด (ของรายการ) แต่จุดตัดทำงานบนชุดข้อมูลขนาดใหญ่ที่ดีที่สุด!

ในบางกรณี (เช่นองค์ประกอบรายการที่ไม่ซ้ำกัน) สามารถใช้การดำเนินการแบบกำหนดได้

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

หรือใช้set.isdisjoint () ,

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

สิ่งนี้จะทำในหนึ่งบรรทัด

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

ฉันรวบรวมวิธีแก้ไขปัญหาต่าง ๆ ที่กล่าวถึงในคำตอบอื่น ๆ และในความคิดเห็นจากนั้นทำการทดสอบความเร็ว เปิดออกเป็นที่เร็วที่สุดก็ยังไม่ได้ชะลอตัวลงมากเมื่อผลที่ได้not set(a).isdisjoint(b)False

ทั้งสามวิ่งทดสอบตัวอย่างเล็ก ๆ ของการกำหนดค่าที่เป็นไปได้ของและa bเวลามีหน่วยเป็นไมโครวินาที

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

ฉันต้องบอกว่าสถานการณ์ของฉันอาจไม่ใช่สิ่งที่คุณกำลังมองหา แต่อาจเป็นทางเลือกให้กับความคิดของคุณ

ฉันลองทั้ง set () และ method ใด ๆ () แต่ยังมีปัญหาเรื่องความเร็ว ดังนั้นฉันจึงจำได้ว่า Raymond Hettinger กล่าวว่าทุกอย่างในไพ ธ อนเป็นพจนานุกรมและใช้ dict ทุกครั้งที่คุณทำได้ นั่นคือสิ่งที่ฉันพยายาม

ฉันใช้ defaultdict กับ int เพื่อระบุผลลัพธ์เชิงลบและใช้รายการในรายการแรกเป็นคีย์สำหรับรายการที่สอง (แปลงเป็น defaultdict) เนื่องจากคุณมีการค้นหาแบบ dict ทันทีคุณจึงรู้ได้ทันทีว่ามีรายการนั้นอยู่ใน defaultdict หรือไม่ ฉันรู้ว่าคุณไม่ได้เปลี่ยนโครงสร้างข้อมูลสำหรับรายการที่สองของคุณเสมอไป แต่ถ้าคุณสามารถทำได้ตั้งแต่เริ่มต้นมันก็จะเร็วขึ้นมาก คุณอาจต้องแปลง list2 (รายการที่ใหญ่ขึ้น) เป็น defaultdict โดยที่ key คือค่าที่เป็นไปได้ที่คุณต้องการตรวจสอบจากรายการเล็ก ๆ และค่าคือ 1 (hit) หรือ 0 (ไม่มี hit, default)

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

ง่าย

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