Kotlin / Kotlin/dataframe

`GroupBy.take()` and other missing functions

Open
#686 4 comments 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
Kotlin
Stars
1.1k
Forks
83
Avg merge
4d 12h
Merged PRs (30d)
30

Description

Currently, we can't do:
```kt
groupedDf
.take(10)
.concat()
```
to only concatenate the values of the first 10 groups. Instead, we'll have to convert to a normal DF first and convert back:
```kt
groupedDf
.toDataFrame().take(10).asGroupBy()
.concat()
```
The only row-based function that's available is `filter(GroupedRowFilter)` which can allow you to write `.filter { it.index() <= 10 }` but seems a bit odd.

Contributor guide

Open the contributing guide

Research direction

Start by reading the GroupBy API around GroupBy.take(), filter(GroupedRowFilter), and the toDataFrame().asGroupBy() conversion shown in the issue. Compare the available grouped operations with the requested row-based behavior and clarify which other missing functions belong in scope. Done means the example can take the first 10 groups before concat(), with tests covering the resulting grouped operation.

Written by the indexing model from the issue text.

Assessment

Tech stack
kotlin
Domain
data
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.