Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEA] Support top_k_by and bottom_k_by #18239

Open
Matt711 opened this issue Mar 11, 2025 · 0 comments
Open

[FEA] Support top_k_by and bottom_k_by #18239

Matt711 opened this issue Mar 11, 2025 · 0 comments
Assignees
Labels
cudf.polars Issues specific to cudf.polars feature request New feature or request Python Affects Python cuDF API.

Comments

@Matt711
Copy link
Contributor

Matt711 commented Mar 11, 2025

Is your feature request related to a problem? Please describe.
We should support both top_k_by and bottom_k_by

df.select(
    pl.all().top_k_by("a", 2).name.suffix("_top_by_a"),
    pl.all().top_k_by("b", 2).name.suffix("_top_by_b"),
)

Describe the solution you'd like
Expose them both in polars (if needed) and then add them to cudf.polars

xref #16223

@Matt711 Matt711 added the feature request New feature or request label Mar 11, 2025
@Matt711 Matt711 self-assigned this Mar 11, 2025
@Matt711 Matt711 added Python Affects Python cuDF API. cudf.polars Issues specific to cudf.polars labels Mar 11, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cudf.polars Issues specific to cudf.polars feature request New feature or request Python Affects Python cuDF API.
Projects
Status: Todo
Development

No branches or pull requests

1 participant