Update wheredataframe.jl

JuliaData · Oct 4, 2020 · 2797e77 · 2797e77
1 parent 5457f29
commit 2797e77
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/wheredataframe/wheredataframe.jl b/src/wheredataframe/wheredataframe.jl
@@ -134,16 +134,17 @@ Base.filter(wdf::WhereDataFrame) = parent(wdf)[rows(wdf), :]
 Base.filter!(wdf::WhereDataFrame) = delete!(parent(wdf), setdiff(1:nrow(parent(wdf)), rows(wdf)))
 Base.delete!(wdf::WhereDataFrame) = delete!(parent(wdf), rows(wdf))
 Base.view(wdf::WhereDataFrame) = view(parent(wdf), rows(wdf), :)
+Base.unique!(wdf::WhereDataFrame, args...) =  delete!(parent(wdf), rows(wdf)[nonunique(view(wdf), args...)])
 # TODO:
 # Base.select
 # Base.select!
 # Base.transform
 # Base.transform!
 
-
-# Functions for which it gives the same thing as view
 DataFrame(wdf::WhereDataFrame; copycols::Bool=true) = DataFrame(view(wdf); copycols = copycols)
 Base.first(df::WhereDataFrame, args...) = first(view(df), args...)
 Base.last(df::WhereDataFrame, args...) = last(view(df), args...)
 DataAPI.describe(wdf::WhereDataFrame, args...; kwargs...) = describe(view(wdf), args...; kwargs...)
-combine(wdf::WhereDataFrame, args...; kwargs...) = combine(view(wdf), args...; kwargs...)
+combine(wdf::WhereDataFrame, args...; kwargs...) = combine(view(wdf), args...; kwargs...)
+Base.unique(wdf::WhereDataFrame, args...) = unique(view(wdf), args...)
+Base.similar(wdf::WhereDataFrame) = similar(view(wdf))