Szathmáry László honlapja @ DEIK

list / set / dict comprehensions

list comprehension

Tanultuk, semmi meglepő nincs benne. A segítségével egy listát tudunk előállítani kompakt módon.

li = [1, 2, 3, 4, 5, 6]
squares = [n*n for n in li] # list comp.
print(squares)

[1, 4, 9, 16, 25, 36]

A list comp. végére mehet egy if feltétel is, amivel egy szűrést tudunk megvalósítani.

set comprehension

Hasonló a list comp.-höz, csak most lista helyett egy halmazt fogunk kompakt módon előállítani. Szintaktikailag annyi a különbség, hogy a "[]" jelek helyett "{}" jeleket használunk.

li = [1, 2, 3, 4, 5, 6]
squares = {n*n for n in li} # set comp.
print(squares)
print(type(squares))

{1, 4, 36, 9, 16, 25}
<class 'set'>

Ha eddig azt csináltuk, hogy list comp.-nel előállítottunk egy listát, majd azt átalakítottuk halmazzá, akkor itt az idő áttérni a set comp.-re.

dict comprehension

Comprehension segítségével szótárat is elő lehet állítani. Szintaktikailag hasonló a set comp.-höz, csak itt az elöl lévő kifejezés egy "kulcs: érték" pár lesz (kettősponttal elválasztva).

li = [1, 2, 3, 4, 5, 6]
squares = {n: n*n for n in li} # dict comp.
print(squares)
print(type(squares))

{1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36}
<class 'dict'>

20170521h