Support for grouping in UUID columns

[Fokko]

Describe the enhancement requested

Python 3.10.16 (main, Dec 3 2024, 17:27:57) [Clang 16.0.0 (clang-1600.0.26.4)]
Type 'copyright', 'credits' or 'license' for more information
IPython 8.31.0 -- An enhanced Interactive Python. Type '?' for help.

In [1]: import pyarrow as pa
 ...: import uuid
 ...: 
 ...: arr_table = pa.Table.from_pydict(
 ...: {
 ...: "uuid": [
 ...: uuid.UUID("00000000-0000-0000-0000-000000000000").bytes,
 ...: uuid.UUID("11111111-1111-1111-1111-111111111111").bytes,
 ...: ],
 ...: },
 ...: schema=pa.schema(
 ...: [
 ...: pa.field("uuid", pa.uuid(), nullable=False),
 ...: ]
 ...: ),
 ...: )
 ...: 
 ...: arr_table.group_by('uuid').aggregate([])
---------------------------------------------------------------------------
ArrowNotImplementedError Traceback (most recent call last)
Cell In[1], line 18
 2 import uuid
 4 arr_table = pa.Table.from_pydict(
 5 {
 6 "uuid": [
 (...)
 15 ),
 16 )
---> 18 arr_table.group_by('uuid').aggregate([])

File /opt/homebrew/lib/python3.10/site-packages/pyarrow/table.pxi:6560, in pyarrow.lib.TableGroupBy.aggregate()

File /opt/homebrew/lib/python3.10/site-packages/pyarrow/acero.py:410, in _group_by(table, aggregates, keys, use_threads)
 404 def _group_by(table, aggregates, keys, use_threads=True):
 406 decl = Declaration.from_sequence([
 407 Declaration("table_source", TableSourceNodeOptions(table)),
 408 Declaration("aggregate", AggregateNodeOptions(aggregates, keys=keys))
 409 ])
--> 410 return decl.to_table(use_threads=use_threads)

File /opt/homebrew/lib/python3.10/site-packages/pyarrow/_acero.pyx:590, in pyarrow._acero.Declaration.to_table()

File /opt/homebrew/lib/python3.10/site-packages/pyarrow/error.pxi:155, in pyarrow.lib.pyarrow_internal_check_status()

File /opt/homebrew/lib/python3.10/site-packages/pyarrow/error.pxi:92, in pyarrow.lib.check_status()

ArrowNotImplementedError: Keys of type extension<arrow.uuid>

Looking at the stacktrace, I think we've need to change something here. The UUID is just a fixed with column under the hood, so I think we can re-use that logic.

Thoughts from the Arrow maintainers?

Component(s)

Python

[pitrou]

This is probably a specific instance of #46671

[github-actions]

This issue has been marked as stale because it has had no activity in the past 365 days. Please remove the stale label or comment below, or this issue will be closed in 14 days. If this improvement is still desired but has no current owner, please add the 'Status: needs champion' label.