I apologize, I picked a poor example of what I want to do. Your suggestion would work for the example I provided, but not for a more complex example. My actual task is something like a &quot;group by&quot; operation along a particular axis (with a known number of groups). <div>

<br></div><div>Let me try again: What I would like to be able to do is to specify some of the iterator dimensions to be handled manually by me. For example lets say I have some kind of a 2d smoothing algorithm. If I start with an array of shape [a,b,c,d] and I&#39;d like to do the 2d smoothing over the 2nd and 3rd dimensions, I&#39;d like to be able to tell nditer to do normal broadcasting and iteration over the 1st and 4th dimensions but leave iteration over the 2nd and 3rd dimensions to me and my algorithm. Each iteration of nditer would give me a 2d array to which I apply my algorithm. This way I could write more arbitrary functions that operate on arrays and support broadcasting.</div>
<div><br></div><div>Is clearer?<br><div>
<br><div class="gmail_quote">On Fri, Sep 30, 2011 at 5:04 PM, Mark Wiebe <span dir="ltr">&lt;<a href="mailto:mwwiebe@gmail.com" target="_blank">mwwiebe@gmail.com</a>&gt;</span> wrote:<br>


<blockquote class="gmail_quote" style="margin:0 0 0 .8ex;border-left:1px #ccc solid;padding-left:1ex"><div class="gmail_quote"><div><div></div><div>On Fri, Sep 30, 2011 at 8:03 AM, John Salvatier <span dir="ltr">&lt;<a href="mailto:jsalvati@u.washington.edu" target="_blank">jsalvati@u.washington.edu</a>&gt;</span> wrote:<br>




<blockquote class="gmail_quote" style="margin:0 0 0 .8ex;border-left:1px #ccc solid;padding-left:1ex">
Using nditer, is it possible to manually handle dimensions  with different lengths? <div><br><div>For example, lets say I had an array A[5, 100] and I wanted to sample every 10 along the second axis so I would end up with an array B[5,10]. Is it possible to do this with nditer, handling the iteration over the second axis manually of course (probably in cython)?</div>







</div><div><br></div><div>I want something like this (modified from <a href="http://docs.scipy.org/doc/numpy/reference/arrays.nditer.html#putting-the-inner-loop-in-cython" target="_blank">http://docs.scipy.org/doc/numpy/reference/arrays.nditer.html#putting-the-inner-loop-in-cython</a>)</div>






<div><br></div><div><div>@cython.boundscheck(False)</div><div>def sum_squares_cy(arr):</div><div>    cdef np.ndarray[double] x</div><div>    cdef np.ndarray[double] y</div><div>    cdef int size</div>
<div>    cdef double value</div><div>    cdef int j</div><div><br></div><div>    axeslist = list(arr.shape)</div><div>    axeslist[1] = -1</div><div><br></div><div>    out = zeros((arr.shape[0], 10))</div><div>    it = np.nditer([arr, out], flags=[&#39;reduce_ok&#39;, &#39;external_loop&#39;,</div>






<div>                                      &#39;buffered&#39;, &#39;delay_bufalloc&#39;],</div>
<div>                op_flags=[[&#39;readonly&#39;], [&#39;readwrite&#39;, &#39;no_broadcast&#39;]],</div><div>                op_axes=[None, axeslist],</div><div>                op_dtypes=[&#39;float64&#39;, &#39;float64&#39;])</div>







<div>    it.operands[1][...] = 0</div><div>    it.reset()</div><div>    for xarr, yarr in it:</div><div>        x = xarr</div><div>        y = yarr</div><div>        size = x.shape[0]</div><div>        j = 0</div><div>        for i in range(size):</div>







<div>           #some magic here involving indexing into x[i] and y[j]</div><div>    return it.operands[1]</div></div><div><br></div><div>Does this make sense? Is it possible to do?</div></blockquote><div><br></div></div>




</div><div>
I&#39;m not sure I understand precisely what you&#39;re asking. Maybe you could reshape A to have shape [5, 10, 10], so that one of those 10&#39;s can match up with the 10 in B, perhaps with the op_axes?</div><div><br></div>





<div>-Mark</div><div> </div><blockquote class="gmail_quote" style="margin:0 0 0 .8ex;border-left:1px #ccc solid;padding-left:1ex">
<br>_______________________________________________<br>
NumPy-Discussion mailing list<br>
<a href="mailto:NumPy-Discussion@scipy.org" target="_blank">NumPy-Discussion@scipy.org</a><br>
<a href="http://mail.scipy.org/mailman/listinfo/numpy-discussion" target="_blank">http://mail.scipy.org/mailman/listinfo/numpy-discussion</a><br>
<br></blockquote></div><br>
<br>_______________________________________________<br>
NumPy-Discussion mailing list<br>
<a href="mailto:NumPy-Discussion@scipy.org" target="_blank">NumPy-Discussion@scipy.org</a><br>
<a href="http://mail.scipy.org/mailman/listinfo/numpy-discussion" target="_blank">http://mail.scipy.org/mailman/listinfo/numpy-discussion</a><br>
<br></blockquote></div><br>
</div>
</div>