List comprehension

List comprehension predstavuje jenoduchý a prehľadný spôsob ako spracovať údaje z dátovej štruktúry -- napríklad zoznamu alebo slovníku prostredníctvom použitia iba jedného riadku kódu.

Vezmime si ako príklad nasledujúci program. Našim cieľom je vytriediť zoznam účastníkov len na tých, ktorí už dosiahli vek 18 rokov:

ages = [5, 14, 23, 18, 40, 21, 8, 10]

adults = []
for age in ages:
  if age >= 18:
    adults.append(age)

print(adults)

Rovnakú vec možno zapísať s pomocou list comprehension nasledovne:

ages = [5, 14, 23, 18, 40, 21, 8, 10]

adults = [age for age in ages if age >= 18]

print(adults)

Kód, ktorý by bol za iných okolností dlhý štyri riadky sme dokázali zapísať na jeden.