Allow for weights in cut(x,[ w,] ngroups)

JuliaData · Sep 18, 2019 · 25ce34b · 25ce34b
1 parent 7d06c34
commit 25ce34b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/extras.jl b/src/extras.jl
@@ -1,4 +1,4 @@
-using Compat.Statistics
+import StatsBase
 
 function fill_refs!(refs::AbstractArray, X::AbstractArray,
                     breaks::AbstractVector, extend::Bool, allow_missing::Bool)
@@ -133,12 +133,17 @@ function cut(x::AbstractArray{T, N}, breaks::AbstractVector;
 end
 
 """
-    cut(x::AbstractArray, ngroups::Integer;
+    cut(x::AbstractArray, [w::AbstractWeights, ]ngroups::Integer;
         labels::Union{AbstractVector{<:AbstractString},Function})
 
 Cut a numeric array into `ngroups` quantiles, determined using
 [`quantile`](@ref).
 """
 cut(x::AbstractArray, ngroups::Integer;
     labels::Union{AbstractVector{<:AbstractString},Function}=default_formatter) =
-    cut(x, Statistics.quantile(x, (1:ngroups-1)/ngroups); extend=true, labels=labels)
+    cut(x, StatsBase.quantile(x, (1:ngroups-1)/ngroups); extend=true, labels=labels)
+
+cut(x::AbstractArray, w::AbstractWeights, ngroups::Integer;
+  labels::Union{AbstractVector{<:AbstractString},Function}=default_formatter) =
+  cut(x, StatsBase.quantile(x, w, (1:ngroups-1)/ngroups); extend=true, labels=labels)
+