[Numpy-discussion] reshaping array question
Robert Kern
robert.kern at gmail.com
Tue Nov 17 11:20:25 EST 2015
On Tue, Nov 17, 2015 at 3:48 PM, Neal Becker <ndbecker2 at gmail.com> wrote:
>
> I have an array of shape
> (7, 24, 2, 1024)
>
> I'd like an array of
> (7, 24, 2048)
>
> such that the elements on the last dimension are interleaving the elements
> from the 3rd dimension
>
> [0,0,0,0] -> [0,0,0]
> [0,0,1,0] -> [0,0,1]
> [0,0,0,1] -> [0,0,2]
> [0,0,1,1] -> [0,0,3]
> ...
>
> What might be the simplest way to do this?
np.transpose(A, (-2, -1)).reshape(A.shape[:-2] + (-1,))
> ------------
> A different question, suppose I just want to stack them
>
> [0,0,0,0] -> [0,0,0]
> [0,0,0,1] -> [0,0,1]
> [0,0,0,2] -> [0,0,2]
> ...
> [0,0,1,0] -> [0,0,1024]
> [0,0,1,1] -> [0,0,1025]
> [0,0,1,2] -> [0,0,1026]
> ...
A.reshape(A.shape[:-2] + (-1,))
--
Robert Kern
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20151117/34634304/attachment.html>
More information about the NumPy-Discussion
mailing list