https://wheretobuyiphone3gchargers.blogspot.com/2024/01/iterating-over-first-d-axes-of-numpy.html