https://github.com/haa-zee/python-sandbox/blob/master/probak/lista_csop…
(itt a hupon ugye nincs syntax highlight és hasonlók?)
Nem pont a módosított feladványhoz illik, erre azt hiszem, lenne valami céleszköz is (array modulban talán), de úgy tűnik, működik. :)