Do you really need the python yield ?

Then you will need to construct an iterator.

Not sure what your list comprehension does, its hard to read.