https://llego.dev/posts/numpy-array-transformations-in-python/