https://mixrolikus.cc/video/ux3ciTUZkCQ/how-to-use-array-iteration-in-python-numpy