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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

//! Text-based visualization functionality
//!
//! This module provides text-based visualization capabilities using the textplots library.
//! These visualizations are lightweight and can be displayed directly in the terminal.

#[cfg(feature = "visualization")]
use std::fs::File;
#[cfg(feature = "visualization")]
use std::io::Write;
use std::path::Path;
#[cfg(feature = "visualization")]
use textplots::{Chart, Plot, Shape};

use crate::error::{PandRSError, Result};
#[cfg(feature = "visualization")]
use crate::vis::config::{OutputFormat, PlotConfig, PlotType};

/// Basic plot function for XY coordinates (text-based)
#[cfg(feature = "visualization")]
pub fn plot_xy<P: AsRef<Path>>(x: &[f32], y: &[f32], path: P, config: PlotConfig) -> Result<()> {
    if x.len() != y.len() {
        return Err(PandRSError::Consistency(
            "X and Y lengths do not match".to_string(),
        ));
    }

    // Do nothing if data is empty
    if x.is_empty() {
        return Err(PandRSError::Empty("No data to plot".to_string()));
    }

    // Create points
    let points: Vec<(f32, f32)> = x.iter().zip(y.iter()).map(|(&x, &y)| (x, y)).collect();

    // Create chart
    let mut chart_string = String::new();
    chart_string.push_str(&format!("=== {} ===\n", config.title));
    chart_string.push_str(&format!(
        "X-axis: {}, Y-axis: {}\n\n",
        config.x_label, config.y_label
    ));

    // Draw plot
    let chart_result = match config.plot_type {
        PlotType::Line => Chart::new(
            config.width as u32,
            config.height as u32,
            x[0],
            x[x.len() - 1],
        )
        .lineplot(&Shape::Lines(&points))
        .to_string(),
        PlotType::Scatter | PlotType::Points => Chart::new(
            config.width as u32,
            config.height as u32,
            x[0],
            x[x.len() - 1],
        )
        .lineplot(&Shape::Points(&points))
        .to_string(),
    };

    chart_string.push_str(&chart_result);

    // Output
    match config.format {
        OutputFormat::Terminal => {
            println!("{}", chart_string);
            Ok(())
        }
        OutputFormat::TextFile => {
            let mut file = File::create(path).map_err(PandRSError::Io)?;
            file.write_all(chart_string.as_bytes())
                .map_err(PandRSError::Io)?;
            Ok(())
        }
    }
}

/// Fallback implementation when visualization is not available
#[cfg(not(feature = "visualization"))]
pub fn plot_xy<P: AsRef<Path>>(
    _x: &[f32],
    _y: &[f32],
    _path: P,
    _config: crate::vis::config::PlotConfig,
) -> Result<()> {
    Err(PandRSError::FeatureNotAvailable(
        "Visualization feature is not enabled. Recompile with --feature visualization".to_string(),
    ))
}