Thanks Hameer for the suggestion. I perhaps should have clarified that this is
a small example for something larger, where I am iterating over a sequence of
arrays, computing some reduction on the first and writing to the second. I am
hoping to keep it all contained within one nditer.
Thanks Er
I am trying to use the nditer to traverse each column of a 2D array, returning
the column as a 1D array. Consulting the docs, I found this example which works
perfectly fine:
In [65]: a = np.arange(6).reshape(2,3)