pandrs 0.3.0

A high-performance DataFrame library for Rust, providing pandas-like API with advanced features including SIMD optimization, parallel processing, and distributed computing capabilities
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

//! # Window Operations for Distributed DataFrames
//!
//! This module provides operations for applying window functions to distributed DataFrames.

use super::core::WindowFunction;
use crate::distributed::dataframe::DistributedDataFrame;
use crate::distributed::execution::{ExecutionPlan, Operation};
use crate::error::Result;

/// Extension trait to add window function capabilities to DistributedDataFrame
pub trait WindowFunctionExt {
    /// Applies window functions to the DataFrame
    fn window(&self, window_functions: &[WindowFunction]) -> Result<DistributedDataFrame>;
}

impl WindowFunctionExt for DistributedDataFrame {
    fn window(&self, window_functions: &[WindowFunction]) -> Result<DistributedDataFrame> {
        // Create a Window operation with the window functions
        let window_functions = window_functions.to_vec();
        let operation = Operation::Window { window_functions };

        if self.is_lazy() {
            let mut new_df = self.clone_empty();
            new_df.add_pending_operation(operation, vec![self.id().to_string()]);
            Ok(new_df)
        } else {
            self.execute_operation(operation, vec![self.id().to_string()])
        }
    }
}