https://llego.dev/posts/numpy-array-indexing-slicing-accessing-python/