https://dosenit.com/python/shape-array-numpy