A more elegant and simple way to traverse the elements of an array in python?

Question

(kk is a 3D array) This is a classic way to "travel around" a 3D array. I am new in python and I sense that there must be a much more elegant way in python