Use np.bincount with the weights optional argument. In your example you would do:
np.bincount(accmap, weights=a)
Answer from Jaime on Stack OverflowUse np.bincount with the weights optional argument. In your example you would do:
np.bincount(accmap, weights=a)
Late to the party, but...
As @Jamie says, for the case of summing, np.bincount is fast and simple. However in the more general case, for other ufuncs such as maximum, you can use the np.ufunc.at method.
I've put together a gist[see link below instead] which encapsulates this in a Matlab-like interface. It also takes advantage of the repeated indexing rules to provide a 'last' and 'first' function, and unlike Matlab, 'mean' is sensibly optimized (calling accumarray with @mean in Matlab is really slow because it runs a non-builtin function for every single group, which is stupid).
Be warned that I haven't particularly tested the gist, but will hopefully update it in future with extra features and bugfixes.
Update May/June-2015: I have reworked my implementation - it is now available as part of ml31415/numpy-groupies and available on PyPi (pip install numpy-groupies). Benchmarks are as follows (see github repo for up-to-date values)...
function pure-py np-grouploop np-ufuncat np-optimised pandas ratio
std 1737.8ms 171.8ms no-impl 7.0ms no-impl 247.1: 24.4: - : 1.0 : -
all 1280.8ms 62.2ms 41.8ms 6.6ms 550.7ms 193.5: 9.4 : 6.3 : 1.0 : 83.2
min 1358.7ms 59.6ms 42.6ms 42.7ms 24.5ms 55.4: 2.4 : 1.7 : 1.7 : 1.0
max 1538.3ms 55.9ms 38.8ms 37.5ms 18.8ms 81.9: 3.0 : 2.1 : 2.0 : 1.0
sum 1532.8ms 62.6ms 40.6ms 1.9ms 20.4ms 808.5: 33.0: 21.4: 1.0 : 10.7
var 1756.8ms 146.2ms no-impl 6.3ms no-impl 279.1: 23.2: - : 1.0 : -
prod 1448.8ms 55.2ms 39.9ms 38.7ms 20.2ms 71.7: 2.7 : 2.0 : 1.9 : 1.0
any 1399.5ms 69.1ms 41.1ms 5.7ms 558.8ms 246.2: 12.2: 7.2 : 1.0 : 98.3
mean 1321.3ms 88.3ms no-impl 4.0ms 20.9ms 327.6: 21.9: - : 1.0 : 5.2
Python 2.7.9, Numpy 1.9.2, Win7 Core i7.
Here we are using 100,000 indices uniformly picked from [0, 1000). Specifically, about 25% of the values are 0 (for use with bool operations), the remainder are uniformly distribuited on [-50,25). Timings are shown for 10 repeats.
- purepy - uses nothing but pure python, relying partly on
itertools.groupby. - np-grouploop - uses
numpyto sort values based onidx, then usessplitto create separate arrays, and then loops over these arrays, running the relevantnumpyfunction for each array. - np-ufuncat - uses the
numpyufunc.atmethod, which is slower than it ought to be - as disuccsed in an issue I created on numpy's github repo. - np-optimisied - uses custom
numpyindexing/other tricks to beat the above two implementations (except formin max prodwhich rely onufunc.at). - pandas -
pd.DataFrame({'idx':idx, 'vals':vals}).groupby('idx').sum()etc.
Note that some of the no-impls may be unwarranted, but I haven't bothered to get them working yet.
As explained on github, accumarray now supports nan-prefixed functions (e.g. nansum) as well as, sort, rsort, and array. It also works with multidimensional indexing.
pandas - Python transcription of Matlab's accumarray - Stack Overflow
What does accumarray([1,2,3], 1) do?
Newest 'accumarray' Questions - Stack Overflow
Analogue to accumarray (matlab) - sparse pooling
I tried with different values for the second argument but it just seems as if the value for the second argument ends up in the last output. Can't make sense out of it. Can someone give me a hint?
Edit. I guess if I vary the first argument to [2, 8] i get more insight, but not sure whether its the right one. Then the output is a 2 by 8 matrix with the second argument being the entry for a_{2,8}. why would one want that? especially compared to the use as a group count function in the first example in its documentation: https://www.mathworks.com/help/matlab/ref/accumarray.html