Tìm kiếm danh sách các đối tượng trong Python

Giả sử tôi đang tạo một lớp đơn giản để hoạt động tương tự như cấu trúc kiểu C, chỉ chứa các phần tử dữ liệu. Tôi đang cố gắng tìm cách tìm kiếm danh sách đối tượng cho các đối tượng có thuộc tính bằng một giá trị nhất định. Dưới đây là một ví dụ nhỏ để minh họa những gì tôi đang cố gắng làm.

Ví dụ:

class Data:
    pass

myList = []

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

Tôi sẽ tìm kiếm danh sách myList như thế nào để xác định xem nó có chứa phần tử với n == 5 hay không?

Tôi đã lên Google và tìm kiếm tài liệu Python và tôi nghĩ rằng tôi có thể làm được điều này với khả năng hiểu danh sách, nhưng tôi không chắc. Nhân tiện, tôi có thể nói thêm rằng tôi đang phải sử dụng Python 2.4.3, vì vậy bất kỳ tính năng gee-whiz 2.6 hoặc 3.x nào mới không khả dụng với tôi.

Bạn có thể nhận được danh sách tất cả các phần tử phù hợp với khả năng hiểu danh sách:

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

Nếu bạn chỉ muốn xác định xem danh sách có chứa bất kỳ phần tử nào phù hợp và thực hiện nó (tương đối) hiệu quả hay không, bạn có thể

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

Đơn giản, Thanh lịch và Mạnh mẽ:

Biểu thức trình tạo trong liên hợp với nội trang… (python 2.5+)

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

Sử dụng any()nội trang Python , được định nghĩa như sau:

any (có thể lặp lại) -> Trả về True nếu bất kỳ phần tử nào của có thể lặp lại là true. Tương đương với:

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

Chỉ để hoàn thiện, chúng ta đừng quên Điều Đơn giản Nhất Có thể Làm được:

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)

Bạn có thể sử dụng inđể tìm kiếm một mục trong bộ sưu tập và hiểu danh sách để trích xuất trường bạn quan tâm. Điều này (hoạt động cho danh sách, bộ, bộ giá trị và bất kỳ thứ gì xác định __contains__hoặc __getitem__).

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

Xem thêm:

• Phương pháp chứa
• Đi vào hoạt động

Bạn nên thêm một __eq__và một __hash__phương thức vào Datalớp của mình , nó có thể kiểm tra xem các __dict__thuộc tính có bằng nhau hay không (cùng thuộc tính) và sau đó nếu giá trị của chúng cũng bằng nhau.

Nếu bạn đã làm điều đó, bạn có thể sử dụng

test = Data()
test.n = 5

found = test in myList

Các inkiểm tra từ khóa nếu testlà trong myList.

Nếu bạn chỉ muốn aa nthuộc tính, Databạn có thể sử dụng:

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

Cân nhắc sử dụng từ điển:

myDict = {}

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

print(5 in myDict)

Một cách khác mà bạn có thể làm là sử dụng hàm next ().

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