ค้นหารายการวัตถุใน Python

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

ตัวอย่างเช่น:

class Data:
    pass

myList = []

for i in range(20):
    data = Data()
    data.n = i
    data.n_squared = i * i
    myList.append(data)

ฉันจะค้นหารายการ myList เพื่อตรวจสอบว่ามีองค์ประกอบที่มี n == 5 ได้อย่างไร

ฉันใช้ Google และค้นหาเอกสาร Python และฉันคิดว่าฉันอาจสามารถทำสิ่งนี้ได้ด้วยความเข้าใจในรายการ แต่ฉันไม่แน่ใจ ฉันอาจเพิ่มว่าฉันต้องใช้ Python 2.4.3 อยู่แล้วดังนั้นฟีเจอร์ gee-whiz 2.6 หรือ 3.x ใหม่ ๆ จึงไม่สามารถใช้ได้กับฉัน

คุณสามารถรับรายการองค์ประกอบที่ตรงกันทั้งหมดพร้อมความเข้าใจในรายการ:

[x for x in myList if x.n == 30]  # list of all elements with .n==30

หากคุณเพียงต้องการตรวจสอบว่ารายการมีองค์ประกอบใดที่ตรงและทำได้ (ค่อนข้าง) มีประสิทธิภาพคุณสามารถทำได้

def contains(list, filter):
    for x in list:
        if filter(x):
            return True
    return False

if contains(myList, lambda x: x.n == 3)  # True if any element has .n==3
    # do stuff

เรียบง่ายสง่างามและทรงพลัง:

นิพจน์เครื่องกำเนิดไฟฟ้าในการเชื่อมต่อกับ builtin ... (python 2.5+)

any(x for x in mylist if x.n == 10)

ใช้ Python any()builtin ซึ่งกำหนดไว้ดังนี้:

any (iterable) -> Return True หากองค์ประกอบใด ๆ ของการทำซ้ำเป็นจริง เทียบเท่ากับ:

def any(iterable):
    for element in iterable:
        if element:
            return True
    return False

เพื่อความสมบูรณ์อย่าลืมสิ่งที่ง่ายที่สุดที่อาจใช้งานได้:

for i in list:
  if i.n == 5:
     # do something with it
     print "YAY! Found one!"

[x for x in myList if x.n == 30]               # list of all matches
[x.n_squared for x in myList if x.n == 30]     # property of matches
any(x.n == 30 for x in myList)                 # if there is any matches
[i for i,x in enumerate(myList) if x.n == 30]  # indices of all matches

def first(iterable, default=None):
  for item in iterable:
    return item
  return default

first(x for x in myList if x.n == 30)          # the first match, if any

filter(lambda x: x.n == 5, myList)

คุณสามารถใช้inเพื่อค้นหาไอเท็มในคอลเลกชั่นและความเข้าใจของรายการเพื่อแยกฟิลด์ที่คุณสนใจสิ่งนี้ (ใช้ได้กับรายการเซ็ตทูเพิลและอะไรก็ตามที่กำหนด__contains__หรือ__getitem__)

if 5 in [data.n for data in myList]:
    print "Found it"

ดูสิ่งนี้ด้วย:

• ประกอบด้วยวิธีการ
• ในการดำเนินการ

คุณควรเพิ่ม a __eq__และ__hash__method ลงในDataคลาสของคุณซึ่งจะตรวจสอบได้ว่า__dict__แอตทริบิวต์เท่ากันหรือไม่ (คุณสมบัติเดียวกัน) จากนั้นค่าของมันจะเท่ากันหรือไม่

หากคุณทำเช่นนั้นคุณสามารถใช้ไฟล์

test = Data()
test.n = 5

found = test in myList

inการตรวจสอบคำหลักถ้าอยู่ในtestmyList

หากคุณต้องการเพียงnคุณสมบัติในDataคุณสามารถใช้:

class Data(object):
    __slots__ = ['n']
    def __init__(self, n):
        self.n = n
    def __eq__(self, other):
        if not isinstance(other, Data):
            return False
        if self.n != other.n:
            return False
        return True
    def __hash__(self):
        return self.n

    myList = [ Data(1), Data(2), Data(3) ]
    Data(2) in myList  #==> True
    Data(5) in myList  #==> False

พิจารณาใช้พจนานุกรม:

myDict = {}

for i in range(20):
    myDict[i] = i * i

print(5 in myDict)

อีกวิธีหนึ่งที่คุณสามารถทำได้คือใช้ฟังก์ชัน next ()

matched_obj = next(x for x in list if x.n == 10)