Skip to content

Add missing Dataframe functions#1472

Draft
timsaucer wants to merge 3 commits intoapache:mainfrom
timsaucer:feat/missing-dataframe-ops
Draft

Add missing Dataframe functions#1472
timsaucer wants to merge 3 commits intoapache:mainfrom
timsaucer:feat/missing-dataframe-ops

Conversation

@timsaucer
Copy link
Copy Markdown
Member

@timsaucer timsaucer commented Mar 31, 2026

Which issue does this PR close?

Closes #1455
Closes #1456

Rationale for this change

These features exist in the upstream repository but have not been exposed in Python.

What changes are included in this PR?

Exposes functions to python
Adds unit tests

Are there any user-facing changes?

Addition only

timsaucer and others added 3 commits March 31, 2026 14:10
Expose upstream DataFusion DataFrame methods that were not yet
available in the Python API. Closes apache#1455.

Set operations:
- except_distinct: set difference with deduplication
- intersect_distinct: set intersection with deduplication
- union_by_name: union matching columns by name instead of position
- union_by_name_distinct: union by name with deduplication

Query:
- distinct_on: deduplicate rows based on specific columns
- sort_by: sort by expressions with ascending order and nulls last

Note: show_limit is already covered by the existing show(num) method.
explain_with_options and with_param_values are deferred as they require
exposing additional types (ExplainOption, ParamValues).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Extend the existing explain() method with an optional format parameter
instead of adding a separate explain_with_options() method. This keeps
the API simple while exposing all upstream ExplainOption functionality.

Available formats: indent (default), tree, pgjson, graphviz.

The ExplainFormat enum is exported from the top-level datafusion module.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Expose remaining DataFrame methods from upstream DataFusion.
Closes apache#1456.

- window(*exprs): apply window function expressions and append results
  as new columns
- unnest_column/unnest_columns: add optional recursions parameter for
  controlling unnest depth via (input_column, output_column, depth)
  tuples

Note: drop_columns is already exposed as the existing drop() method.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add missing DataFrame methods (window, unnest_columns_with_options, drop_columns) Add missing DataFrame methods (set operations and query)

1 participant