https://stats-learn.com/en/python/a-easy-rationalization-of-numpy-axes-with-examples/