从具有对象属性之一的对象列表中获取索引

Question

我有一个对象列表，我试图从我拥有的对象的属性（例如名称）中获取该列表中一个对象的索引。类似于下面的例子：

class Employee:
    def __init__(self, name):
        self.name = name

def add_emp(name):
    employees.append(Employee(name))

employees = []
add_emp('Emp1')

现在我正在尝试在列表 self.employees（此处为 '0'）中获取 'Emp1' 的索引。我在这里试过这个：

print(employees.index(filter(lambda x: x.name == 'Emp1', employees)))

但他告诉我 'ValueError: is not in list'。我需要改变什么或者有更好的选择来处理这个问题？

Answer 1

因为filter() 返回一个过滤器对象，一种方法是将其转换为列表并获取索引为 0 的元素：

print(employees.index(list(filter(lambda x: x.name == 'Emp1', employees))[0]))

但是，最好的方法是使用enumerate()：

def get_employee_index(name):
    for i, e in enumerate(employees):
        if e.name == name:
            return i
    return -1  # for not found employee

输出：

>>> get_employee_index('Emp1')
0

Answer 2

不要搜索过滤器本身，而是搜索过滤器找到的。例如，next(filter(...)) 而不是 filter(...)。

但最好还是使用enumerate:

print(next(i for i, x in enumerate(employees) if x.name == 'Emp1'))

或者您可以创建一个名称列表并要求索引：

print([x.name for x in employees].index('Emp1'))

不过效率较低。