Count instances of each unique integer in a vector in 1 line of code?

后端 未结 5 1013
你的背包
你的背包 2021-01-17 15:08

Is there a slick way to rewrite this Julia function, perhaps using just 1 line of code, without making it much slower? (I just started using Julia. It\'s great!) K

相关标签:
5条回答
  • 2021-01-17 15:28

    I know its old but how about

    [sum(zd .== i) for i in unique(zd)]

    in a short test it performed better than your initial function (time and memory wise).

    Caution: result not sorted!

    0 讨论(0)
  • 2021-01-17 15:32

    I haven't tested the performance, but using the hist function should work:

    hist(zd,0.5:K+0.5)[2]
    

    gives:

    5-element Array{Int64,1}: 1 4 1 0 0

    or, if the zeros are unimportant, just use

    hist(zd)[2]
    3-element Array{Int64,1}:
     1
     4
     1
    
    0 讨论(0)
  • 2021-01-17 15:41

    There are a bunch of counting functions included in the StatsBase.jl package. Your tally function is equivalent to counts(zd, 1:K).

    There are also methods for counting unique elements of types other than integer, too, such as countmap, which returns a dictionary mapping unique values to their number of occurrences.

    0 讨论(0)
  • 2021-01-17 15:46

    Any alternative probably will not be faster. Your loop already does only one pass through the array. Julia loops are fast, and there is no speed advantage to vectorized code, as there is in other languages.

    Have a look at Julia's implementation of the hist function. This is taken directly from the Julia Standard Library:

    function hist(v::AbstractVector, edg::AbstractVector)
        n = length(edg)-1
        h = zeros(Int, n)
        for x in v
            i = searchsortedfirst(edg, x)-1
            if 1 <= i <= n
                h[i] += 1
            end
        end
        edg,h
    end
    

    The "edg" parameter contains the edges of the bins. If we remove that feature, we get exactly the function you wrote.

    EDIT hist has not been available in Julia Base since v0.5.0

    0 讨论(0)
  • 2021-01-17 15:48

    Here http://statsbasejl.readthedocs.org/en/latest/counts.html#countmap

    countmap(x[, wv])
    Return a dictionary that maps distinct values in x to their counts (or total weights).
    
    0 讨论(0)
提交回复
热议问题