https://www.somethingsblog.com/2024/10/20/simplify-numpy-arrays-with-the-power-of-squeeze/