Improve Python code

[hongquan]

• No need to create a list object, which will consume some RAM.
• Take advantage of Python generator and * operator.

[laclouis5]

This decreases memory down to 1.6GB and to 1.25GB with Python 3.11, so this almost halves the memory used by the original program.