BUG (2.0rc0): groupby apply with a UDF changes dtype unexpectedly from double[pyarrow] to float64 #51991
Open
2 of 3 tasks
Labels
Apply
Apply, Aggregate, Transform, Map
Arrow
pyarrow functionality
Bug
Dtype Conversions
Unexpected or buggy dtype conversions
Groupby
pyarrow dtype retention
op with pyarrow dtype -> expect pyarrow result
Pandas version checks
I have checked that this issue has not already been reported.
I have confirmed this bug exists on the latest version of pandas.
I have confirmed this bug exists on the main branch of pandas.
Reproducible Example
Issue Description
Currently having a look at the RC for my current work project. I noticed when using the lambda function with
groupby
together withapply
ortransform
the dtype changes from double[pyarrow] back to float64.Expected Behavior
I'd expect that we consistently get the same datatype
Installed Versions
2.0.0rc0
The text was updated successfully, but these errors were encountered: