https://cerwinvegassalee.blogspot.com/2024/03/numpy-array-loss-of-dimension-when.html