iterdim routine a generator which yields slices of an array

iterdim.py

# found here: http://stackoverflow.com/questions/1589706/iterating-over-arbitrary-dimension-of-numpy-array

import numpy as np

def iterdim(a, axis=0) :
  a = np.asarray(a);
  leading_indices = (slice(None),)*axis
  for i in xrange(a.shape[axis]) :
    yield a[leading_indices+(i,)]
    
# use like:
# asdf = np.random.random((100,10,3))
# for a in iterdim(asdf, axis=1):
#     print a.shape