[Numpy-discussion] permute and sum?

josef.pktd at gmail.com josef.pktd at gmail.com
Mon Feb 28 19:50:53 EST 2011


On Mon, Feb 28, 2011 at 6:50 PM, Angus McMorland <amcmorl at gmail.com> wrote:
> Hi all,
>
> I want to create a transpose of a vector, such that if the same index
> is given in the 'axes' list (as per np.transpose), then the sum of the
> original values sharing the same index is placed in the corresponding
> output index.
>
> For example:
>
> In: data  = np.array([5, 7, 9, 6, 2])
> In: order = np.array([0, 2, 0, 3, 1])
> In: permute_and_sum(data, order)
> Out: array([14, 2, 7, 6])

Not sure I fully understand, but isn't this
np.bincount(order, weights=data)

Josef

> I can obviously do this using the following,
>
> def permute_and_sum(data, order):
>   result = np.zeros(np.max(order) + 1)
>   for val, dest in zip(data, order):
>       result[dest] += val
>   return result
>
> but it seems like there's bound to be a much more elegant method I'm
> not currently seeing. Can anyone point me in the right direction?
>
> Thanks,
>
> Angus.
> --
> AJC McMorland
> Post-doctoral research fellow
> Neurobiology, University of Pittsburgh
> _______________________________________________
> NumPy-Discussion mailing list
> NumPy-Discussion at scipy.org
> http://mail.scipy.org/mailman/listinfo/numpy-discussion
>



More information about the NumPy-Discussion mailing list