https://lusuzi.com/tech/understanding-of-axis-in-numpy/