| |
| <!DOCTYPE html> |
| |
| |
| <html lang="en" data-content_root="../../" data-theme="auto"> |
| |
| <head> |
| <meta charset="utf-8" /> |
| <meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="viewport" content="width=device-width, initial-scale=1" /> |
| |
| <title>Window Functions — Apache DataFusion in Python documentation</title> |
| |
| |
| |
| <script data-cfasync="false"> |
| document.documentElement.dataset.mode = localStorage.getItem("mode") || "auto"; |
| document.documentElement.dataset.theme = localStorage.getItem("theme") || "auto"; |
| </script> |
| <!-- |
| this give us a css class that will be invisible only if js is disabled |
| --> |
| <noscript> |
| <style> |
| .pst-js-only { display: none !important; } |
| |
| </style> |
| </noscript> |
| |
| <!-- Loaded before other Sphinx assets --> |
| <link href="../../_static/styles/theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" /> |
| <link href="../../_static/styles/pydata-sphinx-theme.css?digest=8878045cc6db502f8baf" rel="stylesheet" /> |
| |
| <link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=8f2a1f02" /> |
| <link rel="stylesheet" type="text/css" href="../../_static/mystnb.11b39860a7a0cbfd473a3ad8a317855267ff0bd372690045ca344a6b62be495e.css" /> |
| <link rel="stylesheet" type="text/css" href="../../_static/graphviz.css?v=4ae1632d" /> |
| <link rel="stylesheet" type="text/css" href="../../_static/theme_overrides.css?v=4af573bc" /> |
| |
| <!-- So that users can add custom icons --> |
| <script src="../../_static/scripts/fontawesome.js?digest=8878045cc6db502f8baf"></script> |
| <!-- Pre-loaded scripts that we'll load fully later --> |
| <link rel="preload" as="script" href="../../_static/scripts/bootstrap.js?digest=8878045cc6db502f8baf" /> |
| <link rel="preload" as="script" href="../../_static/scripts/pydata-sphinx-theme.js?digest=8878045cc6db502f8baf" /> |
| |
| <script src="../../_static/documentation_options.js?v=5929fcd5"></script> |
| <script src="../../_static/doctools.js?v=9bcbadda"></script> |
| <script src="../../_static/sphinx_highlight.js?v=dc90522c"></script> |
| <script>DOCUMENTATION_OPTIONS.pagename = 'user-guide/common-operations/windows';</script> |
| <script src="../../_static/toc-toggle.js?v=36375f01"></script> |
| <link rel="icon" href="../../_static/favicon.svg"/> |
| <link rel="index" title="Index" href="../../genindex.html" /> |
| <link rel="search" title="Search" href="../../search.html" /> |
| <link rel="next" title="User-Defined Functions" href="udf-and-udfa.html" /> |
| <link rel="prev" title="Aggregation" href="aggregations.html" /> |
| <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| <meta name="docsearch:language" content="en"/> |
| <meta name="docsearch:version" content="" /> |
| </head> |
| |
| |
| <body data-bs-spy="scroll" data-bs-target=".bd-toc-nav" data-offset="180" data-bs-root-margin="0px 0px -60%" data-default-mode="auto"> |
| |
| |
| |
| <div id="pst-skip-link" class="skip-link d-print-none"><a href="#main-content">Skip to main content</a></div> |
| |
| <div id="pst-scroll-pixel-helper"></div> |
| |
| <button type="button" class="btn rounded-pill" id="pst-back-to-top"> |
| <i class="fa-solid fa-arrow-up"></i>Back to top</button> |
| |
| |
| <dialog id="pst-search-dialog"> |
| |
| <form class="bd-search d-flex align-items-center" |
| action="../../search.html" |
| method="get"> |
| <i class="fa-solid fa-magnifying-glass"></i> |
| <input type="search" |
| class="form-control" |
| name="q" |
| placeholder="Search the docs ..." |
| aria-label="Search the docs ..." |
| autocomplete="off" |
| autocorrect="off" |
| autocapitalize="off" |
| spellcheck="false"/> |
| <span class="search-button__kbd-shortcut"><kbd class="kbd-shortcut__modifier">Ctrl</kbd>+<kbd>K</kbd></span> |
| </form> |
| </dialog> |
| |
| <div class="pst-async-banner-revealer d-none"> |
| <aside id="bd-header-version-warning" class="d-none d-print-none" aria-label="Version warning"></aside> |
| </div> |
| |
| |
| <header class="bd-header navbar navbar-expand-lg bd-navbar d-print-none"> |
| <div class="bd-header__inner bd-page-width"> |
| <button class="pst-navbar-icon sidebar-toggle primary-toggle" aria-label="Site navigation"> |
| <span class="fa-solid fa-bars"></span> |
| </button> |
| |
| |
| <div class="col-lg-3 navbar-header-items__start"> |
| |
| <div class="navbar-item"> |
| |
| |
| |
| |
| |
| <a class="navbar-brand logo" href="../../index.html"> |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| <img src="../../_static/original.svg" class="logo__image only-light" alt="Apache DataFusion in Python"/> |
| <img src="../../_static/original_dark.svg" class="logo__image only-dark pst-js-only" alt="Apache DataFusion in Python"/> |
| |
| |
| </a></div> |
| |
| </div> |
| |
| <div class="col-lg-9 navbar-header-items"> |
| |
| <div class="me-auto navbar-header-items__center"> |
| |
| <div class="navbar-item"> |
| <nav> |
| <ul class="bd-navbar-elements navbar-nav"> |
| |
| <li class="nav-item current active"> |
| <a class="nav-link nav-internal" href="../index.html"> |
| User Guide |
| </a> |
| </li> |
| |
| |
| <li class="nav-item "> |
| <a class="nav-link nav-internal" href="../../contributor-guide/index.html"> |
| Contributor Guide |
| </a> |
| </li> |
| |
| |
| <li class="nav-item "> |
| <a class="nav-link nav-internal" href="../../autoapi/index.html"> |
| API Reference |
| </a> |
| </li> |
| |
| |
| <li class="nav-item "> |
| <a class="nav-link nav-internal" href="../../links.html"> |
| Links |
| </a> |
| </li> |
| |
| </ul> |
| </nav></div> |
| |
| </div> |
| |
| |
| <div class="navbar-header-items__end"> |
| |
| <div class="navbar-item navbar-persistent--container"> |
| |
| |
| <button class="btn search-button-field search-button__button pst-js-only" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip"> |
| <i class="fa-solid fa-magnifying-glass"></i> |
| <span class="search-button__default-text">Search</span> |
| <span class="search-button__kbd-shortcut"><kbd class="kbd-shortcut__modifier">Ctrl</kbd>+<kbd class="kbd-shortcut__modifier">K</kbd></span> |
| </button> |
| </div> |
| |
| |
| <div class="navbar-item"><ul class="navbar-icon-links" |
| aria-label="Icon Links"> |
| <li class="nav-item"> |
| |
| |
| |
| |
| |
| |
| |
| |
| <a href="https://github.com/apache/datafusion-python" title="GitHub" class="nav-link pst-navbar-icon" rel="noopener" target="_blank" data-bs-toggle="tooltip" data-bs-placement="bottom"><i class="fa-brands fa-github fa-lg" aria-hidden="true"></i> |
| <span class="sr-only">GitHub</span></a> |
| </li> |
| <li class="nav-item"> |
| |
| |
| |
| |
| |
| |
| |
| |
| <a href="https://docs.rs/datafusion/latest/datafusion/" title="Rust API docs (docs.rs)" class="nav-link pst-navbar-icon" rel="noopener" target="_blank" data-bs-toggle="tooltip" data-bs-placement="bottom"><i class="fa-brands fa-rust fa-lg" aria-hidden="true"></i> |
| <span class="sr-only">Rust API docs (docs.rs)</span></a> |
| </li> |
| </ul></div> |
| |
| <div class="navbar-item"> |
| |
| <button class="btn btn-sm nav-link pst-navbar-icon theme-switch-button pst-js-only" aria-label="Color mode" data-bs-title="Color mode" data-bs-placement="bottom" data-bs-toggle="tooltip"> |
| <i class="theme-switch fa-solid fa-sun fa-lg" data-mode="light" title="Light"></i> |
| <i class="theme-switch fa-solid fa-moon fa-lg" data-mode="dark" title="Dark"></i> |
| <i class="theme-switch fa-solid fa-circle-half-stroke fa-lg" data-mode="auto" title="System Settings"></i> |
| </button></div> |
| |
| </div> |
| |
| </div> |
| |
| |
| <div class="navbar-persistent--mobile"> |
| |
| <button class="btn search-button-field search-button__button pst-js-only" title="Search" aria-label="Search" data-bs-placement="bottom" data-bs-toggle="tooltip"> |
| <i class="fa-solid fa-magnifying-glass"></i> |
| <span class="search-button__default-text">Search</span> |
| <span class="search-button__kbd-shortcut"><kbd class="kbd-shortcut__modifier">Ctrl</kbd>+<kbd class="kbd-shortcut__modifier">K</kbd></span> |
| </button> |
| </div> |
| |
| |
| |
| <button class="pst-navbar-icon sidebar-toggle secondary-toggle" aria-label="On this page"> |
| <span class="fa-solid fa-outdent"></span> |
| </button> |
| |
| </div> |
| |
| </header> |
| |
| |
| <div class="bd-container"> |
| <div class="bd-container__inner bd-page-width"> |
| |
| |
| |
| <dialog id="pst-primary-sidebar-modal"></dialog> |
| <div id="pst-primary-sidebar" class="bd-sidebar-primary bd-sidebar"> |
| |
| |
| |
| <div class="sidebar-header-items sidebar-primary__section"> |
| |
| |
| <div class="sidebar-header-items__center"> |
| |
| |
| |
| <div class="navbar-item"> |
| <nav> |
| <ul class="bd-navbar-elements navbar-nav"> |
| |
| <li class="nav-item current active"> |
| <a class="nav-link nav-internal" href="../index.html"> |
| User Guide |
| </a> |
| </li> |
| |
| |
| <li class="nav-item "> |
| <a class="nav-link nav-internal" href="../../contributor-guide/index.html"> |
| Contributor Guide |
| </a> |
| </li> |
| |
| |
| <li class="nav-item "> |
| <a class="nav-link nav-internal" href="../../autoapi/index.html"> |
| API Reference |
| </a> |
| </li> |
| |
| |
| <li class="nav-item "> |
| <a class="nav-link nav-internal" href="../../links.html"> |
| Links |
| </a> |
| </li> |
| |
| </ul> |
| </nav></div> |
| |
| |
| </div> |
| |
| |
| |
| <div class="sidebar-header-items__end"> |
| |
| <div class="navbar-item"><ul class="navbar-icon-links" |
| aria-label="Icon Links"> |
| <li class="nav-item"> |
| |
| |
| |
| |
| |
| |
| |
| |
| <a href="https://github.com/apache/datafusion-python" title="GitHub" class="nav-link pst-navbar-icon" rel="noopener" target="_blank" data-bs-toggle="tooltip" data-bs-placement="bottom"><i class="fa-brands fa-github fa-lg" aria-hidden="true"></i> |
| <span class="sr-only">GitHub</span></a> |
| </li> |
| <li class="nav-item"> |
| |
| |
| |
| |
| |
| |
| |
| |
| <a href="https://docs.rs/datafusion/latest/datafusion/" title="Rust API docs (docs.rs)" class="nav-link pst-navbar-icon" rel="noopener" target="_blank" data-bs-toggle="tooltip" data-bs-placement="bottom"><i class="fa-brands fa-rust fa-lg" aria-hidden="true"></i> |
| <span class="sr-only">Rust API docs (docs.rs)</span></a> |
| </li> |
| </ul></div> |
| |
| <div class="navbar-item"> |
| |
| <button class="btn btn-sm nav-link pst-navbar-icon theme-switch-button pst-js-only" aria-label="Color mode" data-bs-title="Color mode" data-bs-placement="bottom" data-bs-toggle="tooltip"> |
| <i class="theme-switch fa-solid fa-sun fa-lg" data-mode="light" title="Light"></i> |
| <i class="theme-switch fa-solid fa-moon fa-lg" data-mode="dark" title="Dark"></i> |
| <i class="theme-switch fa-solid fa-circle-half-stroke fa-lg" data-mode="auto" title="System Settings"></i> |
| </button></div> |
| |
| </div> |
| |
| </div> |
| |
| <div class="sidebar-primary-items__start sidebar-primary__section"> |
| <div class="sidebar-primary-item"> |
| <nav class="bd-docs-nav bd-links" aria-label="Section Navigation"> |
| <p class="bd-links__title" role="heading" aria-level="1">Section Navigation</p> |
| <div class="bd-toc-item navbar-nav"> |
| <ul class="current nav bd-sidenav"> |
| <li class="toctree-l1 current active has-children"><a class="reference internal" href="../index.html">User Guide</a><details open="open"><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul class="current"> |
| <li class="toctree-l2"><a class="reference internal" href="../introduction.html">Introduction</a></li> |
| <li class="toctree-l2"><a class="reference internal" href="../basics.html">Concepts</a></li> |
| <li class="toctree-l2"><a class="reference internal" href="../data-sources.html">Data Sources</a></li> |
| |
| <li class="toctree-l2 has-children"><a class="reference internal" href="../dataframe/index.html">DataFrames</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul> |
| <li class="toctree-l3"><a class="reference internal" href="../dataframe/rendering.html">DataFrame Rendering</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../dataframe/execution-metrics.html">Execution Metrics</a></li> |
| </ul> |
| </details></li> |
| <li class="toctree-l2 current active has-children"><a class="reference internal" href="index.html">Common Operations</a><details open="open"><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul class="current"> |
| <li class="toctree-l3"><a class="reference internal" href="views.html">Registering Views</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="basic-info.html">Basic Operations</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="select-and-filter.html">Column Selections</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="expressions.html">Expressions</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="joins.html">Joins</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="functions.html">Functions</a></li> |
| |
| <li class="toctree-l3"><a class="reference internal" href="spark-functions.html">Spark-Compatible Functions</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="aggregations.html">Aggregation</a></li> |
| <li class="toctree-l3 current active"><a class="current reference internal" href="#">Window Functions</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="udf-and-udfa.html">User-Defined Functions</a></li> |
| </ul> |
| </details></li> |
| <li class="toctree-l2 has-children"><a class="reference internal" href="../io/index.html">IO</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul> |
| <li class="toctree-l3"><a class="reference internal" href="../io/arrow.html">Arrow</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../io/avro.html">Avro</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../io/csv.html">CSV</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../io/json.html">JSON</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../io/parquet.html">Parquet</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../io/table_provider.html">Custom Table Provider</a></li> |
| </ul> |
| </details></li> |
| <li class="toctree-l2"><a class="reference internal" href="../configuration.html">Configuration</a></li> |
| <li class="toctree-l2"><a class="reference internal" href="../distributing-work.html">Distributing work</a></li> |
| <li class="toctree-l2"><a class="reference internal" href="../sql.html">SQL</a></li> |
| <li class="toctree-l2"><a class="reference internal" href="../upgrade-guides.html">Upgrade Guides</a></li> |
| <li class="toctree-l2"><a class="reference internal" href="../ai-coding-assistants.html">Using AI Coding Assistants</a></li> |
| </ul> |
| </details></li> |
| <li class="toctree-l1 has-children"><a class="reference internal" href="../../contributor-guide/index.html">Contributor Guide</a><details open="open"><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul> |
| <li class="toctree-l2"><a class="reference internal" href="../../contributor-guide/introduction.html">Introduction</a></li> |
| <li class="toctree-l2"><a class="reference internal" href="../../contributor-guide/ffi.html">Python Extensions</a></li> |
| </ul> |
| </details></li> |
| <li class="toctree-l1 has-children"><a class="reference internal" href="../../autoapi/index.html">API Reference</a><details open="open"><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul> |
| <li class="toctree-l2 has-children"><a class="reference internal" href="../../autoapi/datafusion/index.html">datafusion</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/catalog/index.html">datafusion.catalog</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/context/index.html">datafusion.context</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/dataframe/index.html">datafusion.dataframe</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/dataframe_formatter/index.html">datafusion.dataframe_formatter</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/expr/index.html">datafusion.expr</a></li> |
| <li class="toctree-l3 has-children"><a class="reference internal" href="../../autoapi/datafusion/functions/index.html">datafusion.functions</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul> |
| <li class="toctree-l4"><a class="reference internal" href="../../autoapi/datafusion/functions/spark/index.html">datafusion.functions.spark</a></li> |
| </ul> |
| </details></li> |
| <li class="toctree-l3 has-children"><a class="reference internal" href="../../autoapi/datafusion/input/index.html">datafusion.input</a><details><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul> |
| <li class="toctree-l4"><a class="reference internal" href="../../autoapi/datafusion/input/base/index.html">datafusion.input.base</a></li> |
| <li class="toctree-l4"><a class="reference internal" href="../../autoapi/datafusion/input/location/index.html">datafusion.input.location</a></li> |
| </ul> |
| </details></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/io/index.html">datafusion.io</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/ipc/index.html">datafusion.ipc</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/object_store/index.html">datafusion.object_store</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/options/index.html">datafusion.options</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/plan/index.html">datafusion.plan</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/record_batch/index.html">datafusion.record_batch</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/substrait/index.html">datafusion.substrait</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/unparser/index.html">datafusion.unparser</a></li> |
| <li class="toctree-l3"><a class="reference internal" href="../../autoapi/datafusion/user_defined/index.html">datafusion.user_defined</a></li> |
| </ul> |
| </details></li> |
| </ul> |
| </details></li> |
| <li class="toctree-l1 has-children"><a class="reference internal" href="../../links.html">Links</a><details open="open"><summary><span class="toctree-toggle" role="presentation"><i class="fa-solid fa-chevron-down"></i></span></summary><ul> |
| <li class="toctree-l2"><a class="reference external" href="https://github.com/apache/datafusion-python">GitHub and Issue Tracker</a></li> |
| <li class="toctree-l2"><a class="reference external" href="https://docs.rs/datafusion/latest/datafusion/">Rust API Docs</a></li> |
| <li class="toctree-l2"><a class="reference external" href="https://github.com/apache/datafusion/blob/main/CODE_OF_CONDUCT.md">Code of Conduct</a></li> |
| <li class="toctree-l2"><a class="reference external" href="https://github.com/apache/datafusion-python/tree/main/examples">Examples</a></li> |
| </ul> |
| </details></li> |
| </ul> |
| </div> |
| </nav></div> |
| </div> |
| |
| |
| <div class="sidebar-primary-items__end sidebar-primary__section"> |
| <div class="sidebar-primary-item"> |
| <div id="ethical-ad-placement" |
| class="flat" |
| data-ea-publisher="readthedocs" |
| data-ea-type="readthedocs-sidebar" |
| data-ea-manual="true"> |
| </div></div> |
| </div> |
| |
| |
| </div> |
| |
| <main id="main-content" class="bd-main" role="main"> |
| |
| |
| <div class="bd-content"> |
| <div class="bd-article-container"> |
| |
| <div class="bd-header-article d-print-none"> |
| <div class="header-article-items header-article__inner"> |
| |
| <div class="header-article-items__start"> |
| |
| <div class="header-article-item"> |
| |
| <nav aria-label="Breadcrumb" class="d-print-none"> |
| <ul class="bd-breadcrumbs"> |
| |
| <li class="breadcrumb-item breadcrumb-home"> |
| <a href="../../index.html" class="nav-link" aria-label="Home"> |
| <i class="fa-solid fa-home"></i> |
| </a> |
| </li> |
| |
| <li class="breadcrumb-item"><a href="../index.html" class="nav-link">User Guide</a></li> |
| |
| |
| <li class="breadcrumb-item"><a href="index.html" class="nav-link">Common Operations</a></li> |
| |
| <li class="breadcrumb-item active" aria-current="page"><span class="ellipsis">Window Functions</span></li> |
| </ul> |
| </nav> |
| </div> |
| |
| </div> |
| |
| |
| </div> |
| </div> |
| |
| |
| |
| |
| <div id="searchbox"></div> |
| <article class="bd-article"> |
| |
| <!--- |
| Licensed to the Apache Software Foundation (ASF) under one |
| or more contributor license agreements. See the NOTICE file |
| distributed with this work for additional information |
| regarding copyright ownership. The ASF licenses this file |
| to you under the Apache License, Version 2.0 (the |
| "License"); you may not use this file except in compliance |
| with the License. You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, |
| software distributed under the License is distributed on an |
| "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| KIND, either express or implied. See the License for the |
| specific language governing permissions and limitations |
| under the License. |
| --> |
| <section id="window-functions"> |
| <span id="id1"></span><h1>Window Functions<a class="headerlink" href="#window-functions" title="Link to this heading">#</a></h1> |
| <p>In this section you will learn about window functions. A window function utilizes values from one or |
| multiple rows to produce a result for each individual row, unlike an aggregate function that |
| provides a single value for multiple rows.</p> |
| <p>The window functions are available in the <a class="reference internal" href="../../autoapi/datafusion/functions/index.html#module-datafusion.functions" title="datafusion.functions"><code class="xref py py-mod docutils literal notranslate"><span class="pre">functions</span></code></a> module.</p> |
| <p>We’ll use the pokemon dataset (from Ritchie Vink) in the following examples.</p> |
| <div class="cell docutils container"> |
| <div class="cell_input docutils container"> |
| <div class="highlight-ipython3 notranslate"><div class="highlight"><pre><span></span><span class="kn">from</span><span class="w"> </span><span class="nn">datafusion</span><span class="w"> </span><span class="kn">import</span> <span class="n">SessionContext</span> |
| <span class="kn">from</span><span class="w"> </span><span class="nn">datafusion</span><span class="w"> </span><span class="kn">import</span> <span class="n">col</span><span class="p">,</span> <span class="n">lit</span> |
| <span class="kn">from</span><span class="w"> </span><span class="nn">datafusion</span><span class="w"> </span><span class="kn">import</span> <span class="n">functions</span> <span class="k">as</span> <span class="n">f</span> |
| |
| <span class="n">ctx</span> <span class="o">=</span> <span class="n">SessionContext</span><span class="p">()</span> |
| <span class="n">df</span> <span class="o">=</span> <span class="n">ctx</span><span class="o">.</span><span class="n">read_csv</span><span class="p">(</span><span class="s2">"pokemon.csv"</span><span class="p">)</span> |
| </pre></div> |
| </div> |
| </div> |
| </div> |
| <p>Here is an example that shows how you can compare each pokemon’s speed to the speed of the |
| previous row in the DataFrame.</p> |
| <div class="cell docutils container"> |
| <div class="cell_input docutils container"> |
| <div class="highlight-ipython3 notranslate"><div class="highlight"><pre><span></span><span class="n">df</span><span class="o">.</span><span class="n">select</span><span class="p">(</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Name"'</span><span class="p">),</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Speed"'</span><span class="p">),</span> |
| <span class="n">f</span><span class="o">.</span><span class="n">lag</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Speed"'</span><span class="p">))</span><span class="o">.</span><span class="n">alias</span><span class="p">(</span><span class="s2">"Previous Speed"</span><span class="p">)</span> |
| <span class="p">)</span> |
| </pre></div> |
| </div> |
| </div> |
| <div class="cell_output docutils container"> |
| <div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>DataFrame() |
| +---------------------------+-------+----------------+ |
| | Name | Speed | Previous Speed | |
| +---------------------------+-------+----------------+ |
| | Bulbasaur | 45 | | |
| | Ivysaur | 60 | 45 | |
| | Venusaur | 80 | 60 | |
| | VenusaurMega Venusaur | 80 | 80 | |
| | Charmander | 65 | 80 | |
| | Charmeleon | 80 | 65 | |
| | Charizard | 100 | 80 | |
| | CharizardMega Charizard X | 100 | 100 | |
| | CharizardMega Charizard Y | 100 | 100 | |
| | Squirtle | 43 | 100 | |
| +---------------------------+-------+----------------+ |
| Data truncated. |
| </pre></div> |
| </div> |
| </div> |
| </div> |
| <section id="setting-parameters"> |
| <h2>Setting Parameters<a class="headerlink" href="#setting-parameters" title="Link to this heading">#</a></h2> |
| <section id="ordering"> |
| <h3>Ordering<a class="headerlink" href="#ordering" title="Link to this heading">#</a></h3> |
| <p>You can control the order in which rows are processed by window functions by providing |
| a list of <code class="docutils literal notranslate"><span class="pre">order_by</span></code> functions for the <code class="docutils literal notranslate"><span class="pre">order_by</span></code> parameter.</p> |
| <div class="cell docutils container"> |
| <div class="cell_input docutils container"> |
| <div class="highlight-ipython3 notranslate"><div class="highlight"><pre><span></span><span class="n">df</span><span class="o">.</span><span class="n">select</span><span class="p">(</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Name"'</span><span class="p">),</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Attack"'</span><span class="p">),</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">),</span> |
| <span class="n">f</span><span class="o">.</span><span class="n">rank</span><span class="p">(</span> |
| <span class="n">partition_by</span><span class="o">=</span><span class="p">[</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">)],</span> |
| <span class="n">order_by</span><span class="o">=</span><span class="p">[</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Attack"'</span><span class="p">)</span><span class="o">.</span><span class="n">sort</span><span class="p">(</span><span class="n">ascending</span><span class="o">=</span><span class="kc">True</span><span class="p">)],</span> |
| <span class="p">)</span><span class="o">.</span><span class="n">alias</span><span class="p">(</span><span class="s2">"rank"</span><span class="p">),</span> |
| <span class="p">)</span><span class="o">.</span><span class="n">sort</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">),</span> <span class="n">col</span><span class="p">(</span><span class="s1">'"Attack"'</span><span class="p">))</span> |
| </pre></div> |
| </div> |
| </div> |
| <div class="cell_output docutils container"> |
| <div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>DataFrame() |
| +------------+--------+--------+------+ |
| | Name | Attack | Type 1 | rank | |
| +------------+--------+--------+------+ |
| | Metapod | 20 | Bug | 1 | |
| | Kakuna | 25 | Bug | 2 | |
| | Caterpie | 30 | Bug | 3 | |
| | Weedle | 35 | Bug | 4 | |
| | Butterfree | 45 | Bug | 5 | |
| | Venonat | 55 | Bug | 6 | |
| | Venomoth | 65 | Bug | 7 | |
| | Paras | 70 | Bug | 8 | |
| | Beedrill | 90 | Bug | 9 | |
| | Parasect | 95 | Bug | 10 | |
| +------------+--------+--------+------+ |
| Data truncated. |
| </pre></div> |
| </div> |
| </div> |
| </div> |
| </section> |
| <section id="partitions"> |
| <h3>Partitions<a class="headerlink" href="#partitions" title="Link to this heading">#</a></h3> |
| <p>A window function can take a list of <code class="docutils literal notranslate"><span class="pre">partition_by</span></code> columns similar to an |
| <a class="reference internal" href="aggregations.html#aggregation"><span class="std std-ref">Aggregation Function</span></a>. This will cause the window values to be evaluated |
| independently for each of the partitions. In the example above, we found the rank of each |
| Pokemon per <code class="docutils literal notranslate"><span class="pre">Type</span> <span class="pre">1</span></code> partitions. We can see the first couple of each partition if we do |
| the following:</p> |
| <div class="cell docutils container"> |
| <div class="cell_input docutils container"> |
| <div class="highlight-ipython3 notranslate"><div class="highlight"><pre><span></span><span class="n">df</span><span class="o">.</span><span class="n">select</span><span class="p">(</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Name"'</span><span class="p">),</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Attack"'</span><span class="p">),</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">),</span> |
| <span class="n">f</span><span class="o">.</span><span class="n">rank</span><span class="p">(</span> |
| <span class="n">partition_by</span><span class="o">=</span><span class="p">[</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">)],</span> |
| <span class="n">order_by</span><span class="o">=</span><span class="p">[</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Attack"'</span><span class="p">)</span><span class="o">.</span><span class="n">sort</span><span class="p">(</span><span class="n">ascending</span><span class="o">=</span><span class="kc">True</span><span class="p">)],</span> |
| <span class="p">)</span><span class="o">.</span><span class="n">alias</span><span class="p">(</span><span class="s2">"rank"</span><span class="p">),</span> |
| <span class="p">)</span><span class="o">.</span><span class="n">filter</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s2">"rank"</span><span class="p">)</span> <span class="o"><</span> <span class="n">lit</span><span class="p">(</span><span class="mi">3</span><span class="p">))</span><span class="o">.</span><span class="n">sort</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">),</span> <span class="n">col</span><span class="p">(</span><span class="s2">"rank"</span><span class="p">))</span> |
| </pre></div> |
| </div> |
| </div> |
| <div class="cell_output docutils container"> |
| <div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>DataFrame() |
| +-----------+--------+----------+------+ |
| | Name | Attack | Type 1 | rank | |
| +-----------+--------+----------+------+ |
| | Metapod | 20 | Bug | 1 | |
| | Kakuna | 25 | Bug | 2 | |
| | Dratini | 64 | Dragon | 1 | |
| | Dragonair | 84 | Dragon | 2 | |
| | Voltorb | 30 | Electric | 1 | |
| | Magnemite | 35 | Electric | 2 | |
| | Clefairy | 45 | Fairy | 1 | |
| | Clefable | 70 | Fairy | 2 | |
| | Mankey | 80 | Fighting | 1 | |
| | Machop | 80 | Fighting | 1 | |
| +-----------+--------+----------+------+ |
| Data truncated. |
| </pre></div> |
| </div> |
| </div> |
| </div> |
| </section> |
| <section id="window-frame"> |
| <h3>Window Frame<a class="headerlink" href="#window-frame" title="Link to this heading">#</a></h3> |
| <p>When using aggregate functions, the Window Frame of defines the rows over which it operates. |
| If you do not specify a Window Frame, the frame will be set depending on the following |
| criteria.</p> |
| <ul class="simple"> |
| <li><p>If an <code class="docutils literal notranslate"><span class="pre">order_by</span></code> clause is set, the default window frame is defined as the rows between |
| unbounded preceding and the current row.</p></li> |
| <li><p>If an <code class="docutils literal notranslate"><span class="pre">order_by</span></code> is not set, the default frame is defined as the rows between unbounded |
| and unbounded following (the entire partition).</p></li> |
| </ul> |
| <p>Window Frames are defined by three parameters: unit type, starting bound, and ending bound.</p> |
| <p>The unit types available are:</p> |
| <ul class="simple"> |
| <li><p>Rows: The starting and ending boundaries are defined by the number of rows relative to the |
| current row.</p></li> |
| <li><p>Range: When using Range, the <code class="docutils literal notranslate"><span class="pre">order_by</span></code> clause must have exactly one term. The boundaries |
| are defined bow how close the rows are to the value of the expression in the <code class="docutils literal notranslate"><span class="pre">order_by</span></code> |
| parameter.</p></li> |
| <li><p>Groups: A “group” is the set of all rows that have equivalent values for all terms in the |
| <code class="docutils literal notranslate"><span class="pre">order_by</span></code> clause.</p></li> |
| </ul> |
| <p>In this example we perform a “rolling average” of the speed of the current Pokemon and the |
| two preceding rows.</p> |
| <div class="cell docutils container"> |
| <div class="cell_input docutils container"> |
| <div class="highlight-ipython3 notranslate"><div class="highlight"><pre><span></span><span class="kn">from</span><span class="w"> </span><span class="nn">datafusion.expr</span><span class="w"> </span><span class="kn">import</span> <span class="n">Window</span><span class="p">,</span> <span class="n">WindowFrame</span> |
| |
| <span class="n">df</span><span class="o">.</span><span class="n">select</span><span class="p">(</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Name"'</span><span class="p">),</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Speed"'</span><span class="p">),</span> |
| <span class="n">f</span><span class="o">.</span><span class="n">avg</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Speed"'</span><span class="p">))</span> |
| <span class="o">.</span><span class="n">over</span><span class="p">(</span><span class="n">Window</span><span class="p">(</span><span class="n">window_frame</span><span class="o">=</span><span class="n">WindowFrame</span><span class="p">(</span><span class="s2">"rows"</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">order_by</span><span class="o">=</span><span class="p">[</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Speed"'</span><span class="p">)]))</span> |
| <span class="o">.</span><span class="n">alias</span><span class="p">(</span><span class="s2">"Previous Speed"</span><span class="p">),</span> |
| <span class="p">)</span> |
| </pre></div> |
| </div> |
| </div> |
| <div class="cell_output docutils container"> |
| <div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>DataFrame() |
| +------------+-------+--------------------+ |
| | Name | Speed | Previous Speed | |
| +------------+-------+--------------------+ |
| | Slowpoke | 15 | 15.0 | |
| | Jigglypuff | 20 | 17.5 | |
| | Geodude | 20 | 18.333333333333332 | |
| | Paras | 25 | 21.666666666666668 | |
| | Grimer | 25 | 23.333333333333332 | |
| | Rhyhorn | 25 | 25.0 | |
| | Snorlax | 30 | 26.666666666666668 | |
| | Metapod | 30 | 28.333333333333332 | |
| | Oddish | 30 | 30.0 | |
| | Parasect | 30 | 30.0 | |
| +------------+-------+--------------------+ |
| Data truncated. |
| </pre></div> |
| </div> |
| </div> |
| </div> |
| </section> |
| <section id="null-treatment"> |
| <h3>Null Treatment<a class="headerlink" href="#null-treatment" title="Link to this heading">#</a></h3> |
| <p>When using aggregate functions as window functions, it is often useful to specify how null values |
| should be treated. In order to do this you need to use the builder function. In future releases |
| we expect this to be simplified in the interface.</p> |
| <p>One common usage for handling nulls is the case where you want to find the last value up to the |
| current row. In the following example we demonstrate how setting the null treatment to ignore |
| nulls will fill in with the value of the most recent non-null row. To do this, we also will set |
| the window frame so that we only process up to the current row.</p> |
| <p>In this example, we filter down to one specific type of Pokemon that does have some entries in |
| it’s <code class="docutils literal notranslate"><span class="pre">Type</span> <span class="pre">2</span></code> column that are null.</p> |
| <div class="cell docutils container"> |
| <div class="cell_input docutils container"> |
| <div class="highlight-ipython3 notranslate"><div class="highlight"><pre><span></span><span class="kn">from</span><span class="w"> </span><span class="nn">datafusion.common</span><span class="w"> </span><span class="kn">import</span> <span class="n">NullTreatment</span> |
| |
| <span class="n">df</span><span class="o">.</span><span class="n">filter</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">)</span> <span class="o">==</span> <span class="n">lit</span><span class="p">(</span><span class="s2">"Bug"</span><span class="p">))</span><span class="o">.</span><span class="n">select</span><span class="p">(</span> |
| <span class="s1">'"Name"'</span><span class="p">,</span> |
| <span class="s1">'"Type 2"'</span><span class="p">,</span> |
| <span class="n">f</span><span class="o">.</span><span class="n">last_value</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Type 2"'</span><span class="p">))</span> |
| <span class="o">.</span><span class="n">over</span><span class="p">(</span> |
| <span class="n">Window</span><span class="p">(</span> |
| <span class="n">window_frame</span><span class="o">=</span><span class="n">WindowFrame</span><span class="p">(</span><span class="s2">"rows"</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> |
| <span class="n">order_by</span><span class="o">=</span><span class="p">[</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Speed"'</span><span class="p">)],</span> |
| <span class="n">null_treatment</span><span class="o">=</span><span class="n">NullTreatment</span><span class="o">.</span><span class="n">IGNORE_NULLS</span><span class="p">,</span> |
| <span class="p">)</span> |
| <span class="p">)</span> |
| <span class="o">.</span><span class="n">alias</span><span class="p">(</span><span class="s2">"last_wo_null"</span><span class="p">),</span> |
| <span class="n">f</span><span class="o">.</span><span class="n">last_value</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Type 2"'</span><span class="p">))</span> |
| <span class="o">.</span><span class="n">over</span><span class="p">(</span> |
| <span class="n">Window</span><span class="p">(</span> |
| <span class="n">window_frame</span><span class="o">=</span><span class="n">WindowFrame</span><span class="p">(</span><span class="s2">"rows"</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> |
| <span class="n">order_by</span><span class="o">=</span><span class="p">[</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Speed"'</span><span class="p">)],</span> |
| <span class="n">null_treatment</span><span class="o">=</span><span class="n">NullTreatment</span><span class="o">.</span><span class="n">RESPECT_NULLS</span><span class="p">,</span> |
| <span class="p">)</span> |
| <span class="p">)</span> |
| <span class="o">.</span><span class="n">alias</span><span class="p">(</span><span class="s2">"last_with_null"</span><span class="p">),</span> |
| <span class="p">)</span> |
| </pre></div> |
| </div> |
| </div> |
| <div class="cell_output docutils container"> |
| <div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>DataFrame() |
| +------------+--------+--------------+----------------+ |
| | Name | Type 2 | last_wo_null | last_with_null | |
| +------------+--------+--------------+----------------+ |
| | Paras | Grass | Grass | Grass | |
| | Metapod | | Grass | | |
| | Parasect | Grass | Grass | Grass | |
| | Kakuna | Poison | Poison | Poison | |
| | Caterpie | | Poison | | |
| | Venonat | Poison | Poison | Poison | |
| | Weedle | Poison | Poison | Poison | |
| | Butterfree | Flying | Flying | Flying | |
| | Beedrill | Poison | Poison | Poison | |
| | Pinsir | | Poison | | |
| +------------+--------+--------------+----------------+ |
| Data truncated. |
| </pre></div> |
| </div> |
| </div> |
| </div> |
| </section> |
| </section> |
| <section id="aggregate-functions"> |
| <h2>Aggregate Functions<a class="headerlink" href="#aggregate-functions" title="Link to this heading">#</a></h2> |
| <p>You can use any <a class="reference internal" href="aggregations.html#aggregation"><span class="std std-ref">Aggregation Function</span></a> as a window function. Here |
| is an example that shows how to compare each pokemons’s attack power with the average attack |
| power in its <code class="docutils literal notranslate"><span class="pre">"Type</span> <span class="pre">1"</span></code> using the <a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.avg" title="datafusion.functions.avg"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.avg()</span></code></a> function.</p> |
| <div class="cell docutils container"> |
| <div class="cell_input docutils container"> |
| <div class="highlight-ipython3 notranslate"><div class="highlight"><pre><span></span><span class="n">df</span><span class="o">.</span><span class="n">select</span><span class="p">(</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Name"'</span><span class="p">),</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Attack"'</span><span class="p">),</span> |
| <span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">),</span> |
| <span class="n">f</span><span class="o">.</span><span class="n">avg</span><span class="p">(</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Attack"'</span><span class="p">))</span><span class="o">.</span><span class="n">over</span><span class="p">(</span> |
| <span class="n">Window</span><span class="p">(</span> |
| <span class="n">window_frame</span><span class="o">=</span><span class="n">WindowFrame</span><span class="p">(</span><span class="s2">"rows"</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span> <span class="kc">None</span><span class="p">),</span> |
| <span class="n">partition_by</span><span class="o">=</span><span class="p">[</span><span class="n">col</span><span class="p">(</span><span class="s1">'"Type 1"'</span><span class="p">)],</span> |
| <span class="p">)</span> |
| <span class="p">)</span><span class="o">.</span><span class="n">alias</span><span class="p">(</span><span class="s2">"Average Attack"</span><span class="p">),</span> |
| <span class="p">)</span> |
| </pre></div> |
| </div> |
| </div> |
| <div class="cell_output docutils container"> |
| <div class="output text_plain highlight-myst-ansi notranslate"><div class="highlight"><pre><span></span>DataFrame() |
| +-------------------+--------+--------+----------------+ |
| | Name | Attack | Type 1 | Average Attack | |
| +-------------------+--------+--------+----------------+ |
| | Dragonite | 134 | Dragon | 94.0 | |
| | Dratini | 64 | Dragon | 94.0 | |
| | Dragonair | 84 | Dragon | 94.0 | |
| | Gastly | 35 | Ghost | 53.75 | |
| | Haunter | 50 | Ghost | 53.75 | |
| | Gengar | 65 | Ghost | 53.75 | |
| | GengarMega Gengar | 65 | Ghost | 53.75 | |
| | Jynx | 50 | Ice | 67.5 | |
| | Articuno | 85 | Ice | 67.5 | |
| | Kabuto | 80 | Rock | 87.5 | |
| +-------------------+--------+--------+----------------+ |
| Data truncated. |
| </pre></div> |
| </div> |
| </div> |
| </div> |
| </section> |
| <section id="available-functions"> |
| <h2>Available Functions<a class="headerlink" href="#available-functions" title="Link to this heading">#</a></h2> |
| <p>The possible window functions are:</p> |
| <ol class="arabic simple"> |
| <li><dl class="simple myst"> |
| <dt>Rank Functions</dt><dd><ul class="simple"> |
| <li><p><a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.rank" title="datafusion.functions.rank"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.rank()</span></code></a></p></li> |
| <li><p><a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.dense_rank" title="datafusion.functions.dense_rank"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.dense_rank()</span></code></a></p></li> |
| <li><p><a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.ntile" title="datafusion.functions.ntile"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.ntile()</span></code></a></p></li> |
| <li><p><a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.row_number" title="datafusion.functions.row_number"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.row_number()</span></code></a></p></li> |
| </ul> |
| </dd> |
| </dl> |
| </li> |
| <li><dl class="simple myst"> |
| <dt>Analytical Functions</dt><dd><ul class="simple"> |
| <li><p><a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.cume_dist" title="datafusion.functions.cume_dist"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.cume_dist()</span></code></a></p></li> |
| <li><p><a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.percent_rank" title="datafusion.functions.percent_rank"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.percent_rank()</span></code></a></p></li> |
| <li><p><a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.lag" title="datafusion.functions.lag"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.lag()</span></code></a></p></li> |
| <li><p><a class="reference internal" href="../../autoapi/datafusion/functions/index.html#datafusion.functions.lead" title="datafusion.functions.lead"><code class="xref py py-func docutils literal notranslate"><span class="pre">datafusion.functions.lead()</span></code></a></p></li> |
| </ul> |
| </dd> |
| </dl> |
| </li> |
| <li><dl class="simple myst"> |
| <dt>Aggregate Functions</dt><dd><ul class="simple"> |
| <li><p>All <a class="reference internal" href="aggregations.html#aggregation"><span class="std std-ref">Aggregation Functions</span></a> can be used as window functions.</p></li> |
| </ul> |
| </dd> |
| </dl> |
| </li> |
| </ol> |
| </section> |
| <section id="user-defined-window-functions"> |
| <h2>User-Defined Window Functions<a class="headerlink" href="#user-defined-window-functions" title="Link to this heading">#</a></h2> |
| <p>You can ship custom window functions to the engine by subclassing |
| <a class="reference internal" href="../../autoapi/datafusion/user_defined/index.html#datafusion.user_defined.WindowEvaluator" title="datafusion.user_defined.WindowEvaluator"><code class="xref py py-class docutils literal notranslate"><span class="pre">WindowEvaluator</span></code></a> and registering it |
| via <a class="reference internal" href="../../autoapi/datafusion/index.html#datafusion.udwf" title="datafusion.udwf"><code class="xref py py-func docutils literal notranslate"><span class="pre">udwf()</span></code></a>. See <a class="reference internal" href="../../autoapi/datafusion/user_defined/index.html#module-datafusion.user_defined" title="datafusion.user_defined"><code class="xref py py-mod docutils literal notranslate"><span class="pre">datafusion.user_defined</span></code></a> |
| for the evaluator interface and worked examples.</p> |
| <div class="admonition note"> |
| <p class="admonition-title">Note</p> |
| <p>Serialization</p> |
| <p>Python window UDFs travel inline inside pickled or |
| <a class="reference internal" href="../../autoapi/datafusion/expr/index.html#datafusion.expr.Expr.to_bytes" title="datafusion.expr.Expr.to_bytes"><code class="xref py py-meth docutils literal notranslate"><span class="pre">to_bytes()</span></code></a>-serialized expressions — |
| the evaluator class is captured by value via <code class="xref py py-mod docutils literal notranslate"><span class="pre">cloudpickle</span></code>, so |
| worker processes do not need to pre-register the UDF. Any names the |
| evaluator resolves via <code class="docutils literal notranslate"><span class="pre">import</span></code> are captured <strong>by reference</strong> and |
| must be importable on the receiving worker. See |
| <a class="reference internal" href="../../autoapi/datafusion/ipc/index.html#module-datafusion.ipc" title="datafusion.ipc"><code class="xref py py-mod docutils literal notranslate"><span class="pre">datafusion.ipc</span></code></a> for the full IPC model and security caveats.</p> |
| </div> |
| </section> |
| </section> |
| |
| |
| </article> |
| |
| |
| |
| |
| |
| <footer class="prev-next-footer d-print-none"> |
| |
| <div class="prev-next-area"> |
| <a class="left-prev" |
| href="aggregations.html" |
| title="previous page"> |
| <i class="fa-solid fa-angle-left"></i> |
| <div class="prev-next-info"> |
| <p class="prev-next-subtitle">previous</p> |
| <p class="prev-next-title">Aggregation</p> |
| </div> |
| </a> |
| <a class="right-next" |
| href="udf-and-udfa.html" |
| title="next page"> |
| <div class="prev-next-info"> |
| <p class="prev-next-subtitle">next</p> |
| <p class="prev-next-title">User-Defined Functions</p> |
| </div> |
| <i class="fa-solid fa-angle-right"></i> |
| </a> |
| </div> |
| </footer> |
| |
| </div> |
| |
| |
| |
| <dialog id="pst-secondary-sidebar-modal"></dialog> |
| <div id="pst-secondary-sidebar" class="bd-sidebar-secondary bd-toc"><div class="sidebar-secondary-items sidebar-secondary__inner"> |
| |
| |
| <div class="sidebar-secondary-item"> |
| <div |
| id="pst-page-navigation-heading-2" |
| class="page-toc tocsection onthispage"> |
| <i class="fa-solid fa-list"></i> On this page |
| </div> |
| <nav class="bd-toc-nav page-toc" aria-labelledby="pst-page-navigation-heading-2"> |
| <ul class="visible nav section-nav flex-column"> |
| <li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#setting-parameters">Setting Parameters</a><ul class="visible nav section-nav flex-column"> |
| <li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#ordering">Ordering</a></li> |
| <li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#partitions">Partitions</a></li> |
| <li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#window-frame">Window Frame</a></li> |
| <li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#null-treatment">Null Treatment</a></li> |
| </ul> |
| </li> |
| <li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#aggregate-functions">Aggregate Functions</a></li> |
| <li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#available-functions">Available Functions</a></li> |
| <li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#user-defined-window-functions">User-Defined Window Functions</a></li> |
| </ul> |
| </nav></div> |
| |
| </div></div> |
| |
| |
| </div> |
| <footer class="bd-footer-content"> |
| |
| </footer> |
| |
| </main> |
| </div> |
| </div> |
| |
| <!-- Scripts loaded after <body> so the DOM is not blocked --> |
| <script defer src="../../_static/scripts/bootstrap.js?digest=8878045cc6db502f8baf"></script> |
| <script defer src="../../_static/scripts/pydata-sphinx-theme.js?digest=8878045cc6db502f8baf"></script> |
| |
| <!-- Based on pydata_sphinx_theme/footer.html --> |
| <footer class="footer mt-5 mt-md-0"> |
| <div class="container"> |
| |
| <div class="footer-item"> |
| <p>Apache Arrow DataFusion, Arrow DataFusion, Apache, the Apache feather logo, and the Apache Arrow DataFusion project logo</p> |
| <p>are either registered trademarks or trademarks of The Apache Software Foundation in the United States and other countries.</p> |
| </div> |
| </div> |
| </footer> |
| |
| |
| </body> |
| </html> |