(self)
| 2655 | # GroupBy Methods |
| 2656 | |
| 2657 | def _to_masked(self): |
| 2658 | pa_dtype = self._pa_array.type |
| 2659 | |
| 2660 | if pa.types.is_floating(pa_dtype) or pa.types.is_integer(pa_dtype): |
| 2661 | na_value = 1 |
| 2662 | elif pa.types.is_boolean(pa_dtype): |
| 2663 | na_value = True |
| 2664 | else: |
| 2665 | raise NotImplementedError |
| 2666 | |
| 2667 | dtype = _arrow_dtype_mapping()[pa_dtype] |
| 2668 | mask = self.isna() |
| 2669 | arr = self.to_numpy(dtype=dtype.numpy_dtype, na_value=na_value) |
| 2670 | return dtype.construct_array_type()(arr, mask) |
| 2671 | |
| 2672 | def _groupby_op( |
| 2673 | self, |
no test coverage detected