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