numpy: frequencies
Tim Hochberg
tim.hochberg at ieee.org
Sat Nov 18 18:41:39 CET 2006
Filip Wasilewski wrote:
> robert wrote:
>> I have an integer array with values limited to range(a,b) like:
>>
>> ia=array([1,2,3,3,3,4,...2,0,1])
>>
>> and want to speedly count the frequencies of the integers into get a density matrix.
>> Is this possible without looping?
>
> See numpy.bincount (for integers >= 0) if you mean 'without writing
> looping code in Python' or please specify your question.
>
>> Question 2: is it possible to compute a "moving maximum" without python looping
>>
>> ia=array([4,2,1,5,3,2,2,0,1,1])
>> -> mvmax(ia,3) ->
>> [4,4,4,5,5,5,3,2,2,1])
>
> I haven't seen a ready solution but this can be easily converted into
> Pyrex/C looping.
I don't know a way to avoid looping entirely, but there are ways that
you can loop over the width of the window (in this case 3) rather than
over the entire array. Since the window width is generally small
compared to the array, this will probably be fast enough. The tricky
part is to get the value right at the edges, since what you do there
depends on what boundary conditions you apply.
The general idea is this:
result = ia[n-1:]
for i in range(n-1):
numpy.maximum(result, ia[i:-n+i], result)
This punts on dealing with the ends (and I haven't tested this version),
but should give you the idea.
-tim
More information about the Python-list
mailing list