https://cerwinvegassalee.blogspot.com/2023/04/optimization-of-python-code-using-numpy.html