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