Skip to content

Window Functions

Functions for window operations and analytical processing.

window

Window dataclass 

Window(
    partition_by: str, order_by: str, desc: bool = False
)

Represents a window specification for SQL window functions.

window 

window(
    partition_by: str, order_by: str, desc: bool = False
) -> Window

Defines a window specification for SQL window functions.

The window function specifies how to partition and order the result set for the associated window function. It is used to define the scope of the rows that the window function will operate on.

Parameters:

  • partition_by (str) –

    The column name by which to partition the result set. Rows with the same value in the partition column will be grouped together for the window function.

  • order_by (str) –

    The column name by which to order the rows within each partition. This determines the sequence in which the window function is applied.

  • desc (bool, default: False ) –

    If True, the rows will be ordered in descending order. Defaults to False, which orders the rows in ascending order.

Returns:

  • Window ( Window ) –

    A Window object representing the window specification.

Example 
window = func.window(partition_by="signal.category", order_by="created_at")
dc.mutate(
    row_number=func.row_number().over(window),
)
Source code in datachain/func/window.py 
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
def window(partition_by: str, order_by: str, desc: bool = False) -> Window:
    """
    Defines a window specification for SQL window functions.

    The `window` function specifies how to partition and order the result set
    for the associated window function. It is used to define the scope of the rows
    that the window function will operate on.

    Args:
        partition_by (str): The column name by which to partition the result set.
            Rows with the same value in the partition column will be grouped together
            for the window function.
        order_by (str): The column name by which to order the rows within
            each partition. This determines the sequence in which the window function
            is applied.
        desc (bool, optional): If True, the rows will be ordered in descending order.
            Defaults to False, which orders the rows in ascending order.

    Returns:
        Window: A `Window` object representing the window specification.

    Example:
        ```py
        window = func.window(partition_by="signal.category", order_by="created_at")
        dc.mutate(
            row_number=func.row_number().over(window),
        )
        ```
    """
    return Window(
        ColumnMeta.to_db_name(partition_by),
        ColumnMeta.to_db_name(order_by),
        desc,
    )