Re: [Numpy-discussion] Non-Zero Sub-Matrix

2010-09-10 Thread Pauli Virtanen
Fri, 10 Sep 2010 14:35:46 +0200, Radek Machulka wrote: > Thanks, but... > x = array([[0,0,0,0],[0,1,0,0],[0,0,1,1],[0,0,0,0]]) x > array([[0, 0, 0, 0], >[0, 1, 0, 0], >[0, 0, 1, 1], >[0, 0, 0, 0]]) i, j = x.any(0).nonzero()[0], x.any(1).nonzero()[0] Should be

Re: [Numpy-discussion] Non-Zero Sub-Matrix

2010-09-10 Thread Radek Machulka
Thanks, but... >>> x = array([[0,0,0,0],[0,1,0,0],[0,0,1,1],[0,0,0,0]]) >>> x array([[0, 0, 0, 0], [0, 1, 0, 0], [0, 0, 1, 1], [0, 0, 0, 0]]) >>> i, j = x.any(0).nonzero()[0], x.any(1).nonzero()[0] >>> x[i[:,None], j[None,:]] array([[1, 0], [0, 1], [0, 0]]) R.

Re: [Numpy-discussion] Non-Zero Sub-Matrix

2010-09-10 Thread Pauli Virtanen
Fri, 10 Sep 2010 11:46:47 +0200, Radek Machulka wrote: > I have array (numpy.ndarray object) with non-zero elements cumulated > 'somewhere' (like a array([[0,0,0,0],[0,1,1,0],[0,0,1,0],[0,0,0,0]])) > and I need sub-array with just non-zero elements (array([[1,1],[0,1]])). > I can do this with itera

[Numpy-discussion] Non-Zero Sub-Matrix

2010-09-10 Thread Radek Machulka
Hi Folks, I have array (numpy.ndarray object) with non-zero elements cumulated 'somewhere' (like a array([[0,0,0,0],[0,1,1,0],[0,0,1,0],[0,0,0,0]])) and I need sub-array with just non-zero elements (array([[1,1],[0,1]])). I can do this with iterating throught an array, but I also found some magic