当在 Python 中给出类属性之一的值时，如何访问列表中的类对象？

Question

我有一个类对象列表，我想知道如何在列表中打印一个特定的类对象，给它一个属性的值。

这是我的代码：

class Person:
  def __init__(self, name, age):
    self.name = name
    self.age = age

lst = [Person("A",1), Person("B",2), Person("C", 3)]

inputAge = int(input("Age: "))

if inputAge == 2:
  print() #this should print the class object with age 2 from the list 

我不知道我应该在print()代码中放什么。

Answer 1

这是您可以找到具有特定属性的列表的第一个元素的方法（前提条件：列表的所有对象都具有该属性）：

obj = next((x for x in lst if x.age == inputAge), None)

如果在列表中找不到这样的对象，则obj为None 。

找到对象后，您可以对它做任何您想做的事情，包括打印（ print(obj) ）。

如果您希望对象的打印有意义，则必须执行以下操作：

class Person:
  def __init__(self, name, age):
    self.name = name
    self.age = age
  def __str__(self):
    return 'name: ' + self.name + ' age: ' + self.age

这是完整的例子：

class Person:
  def __init__(self, name, age):
    self.name = name
    self.age = age
  def __str__(self):
    return 'name: ' + self.name + ' age: ' + self.age

lst = [Person("A",1), Person("B",2), Person("C", 3)]

inputAge = int(input("Age: "))
obj = next((x for x in lst if x.age == inputAge), None)
print(obj) # prints the object with age = inputAge