https://llego.dev/posts/numpy-changing-array-shape/