queryverse / queryverse/Query.jl
`@where` does not preserve CategoricalArray
Open
Nobody has claimed this yet.
discussion
- Dominant language
- Julia
- Stars
- 403
- Forks
- 48
- Avg merge
- 3d 6h
- Merged PRs (30d)
- 6
Description
Julia v0.6, DataFrames, CategoricalArray and Query masters
julia> using DataFrames, Query
julia> df = DataFrame(a = categorical(["a", "b", "a", "c"]))
4×1 DataFrames.DataFrame
│ Row │ a │
├─────┼───┤
│ 1 │ a │
│ 2 │ b │
│ 3 │ a │
│ 4 │ c │
julia> typeof(df[:a])
CategoricalArrays.CategoricalArray{String,1,UInt32,String,Union{}}
julia> dff = df |> @where(_.a != "b") |> DataFrame
3×1 DataFrames.DataFrame
│ Row │ a │
├─────┼───┤
│ 1 │ a │
│ 2 │ a │
│ 3 │ c │
julia> typeof(dff[:a])
Array{CategoricalArrays.CategoricalValue{String,UInt32},1}
IIUC Array{CategoricalValue} is suboptimal in terms of storage. Also CategoricalArrays API like levels(dff[:a]) is not available.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Reproduce the issue with the Julia commands in the report, using DataFrames, CategoricalArray, and Query. Trace the @where pipeline to find where the categorical column changes representation; done means the filtered result retains a CategoricalArray and supports levels(dff[:a]) as expected.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- julia
- Domain
- data
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100