https://remp3indir.net/vectorize-for-loop-python/