Skip to content

ENH: Allow numba aggregations to return non-float64 results #44952

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.

Sign up for GitHub

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

Jump to bottom
Closed
mroeschke opened this issue Dec 18, 2021 · 0 comments
Closed

ENH: Allow numba aggregations to return non-float64 results #44952

mroeschke opened this issue Dec 18, 2021 · 0 comments
Assignees
@lithomas1
Labels
Enhancement numba numba-accelerated operations Reduction Operations sum, mean, min, max, etc.

Comments

@mroeschke
Copy link
Member

For example, groupby.sum can return an int64 output given an int64 input

In [2]: pd.DataFrame(range(2)).groupby(level=0).sum()
Out[2]:
   0
0  0
1  1

To enable this for our numba aggregations, I think we would need to specify specific signatures.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@lithomas1 lithomas1

Labels
Enhancement numba numba-accelerated operations Reduction Operations sum, mean, min, max, etc.
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@mroeschke @lithomas1