Python - zwrócenie listy obiektów o szukanej wartości pola

Question

Hej

Mam taką klasę i listę obiektów tej klasy (niech nazywa się np. cars_list)

 

class Car:
    def __init__(self, id, model, year_of_prod):
        self.id = id
        self.model = model
        self.year_of_prod = year_of_prod

 

Czy jest jakiś prosty i szybki sposób aby z tej listy samochodów (cars_list) wyciągnąć drugą listę samochodów o konkretnej wartości pola np. -> wszystkie samochody dla których year_of_prod wynosi 2001

Wiem że mogę zrobić funkcję z pętlą i sprawdzać po kolei i przypisywać do nowej listy, ale zapewne można to zrobić ładniej (np. w C# użyłbym Linq)

Answer 1

Najłatwiej zwykłym filtrem:

#!/usr/bin/env python3

class Car:
    def __init__(self, id, model, year_of_prod):
        self.id = id
        self.model = model
        self.year_of_prod = year_of_prod

cars_list = [ Car(1, "Ford Mustand", 1983), Car(2, "Fiat 126P", 1981),
        Car(3, "WV Passat", 2001) ]

# Samochody wyprodukowane po 1982
result_cars = filter(lambda c: c.year_of_prod > 1982, cars_list)

for car in result_cars:
    print(car.model, car.year_of_prod)

 

Comments

Alternatywnie:

[car for car in cars_list if car.year_of_prod > 1982]