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

//! Helper functions for functions2 module

use crate::core::error::Result;
use crate::dataframe::base::DataFrame;
use crate::series::Series;

/// Helper function for selecting rows by indices
pub(super) fn select_rows_by_indices(df: &DataFrame, indices: &[usize]) -> Result<DataFrame> {
    let mut new_df = DataFrame::new();
    for col_name in df.column_names() {
        if let Ok(values) = df.get_column_numeric_values(&col_name) {
            let selected: Vec<f64> = indices
                .iter()
                .filter_map(|&i| values.get(i).copied())
                .collect();
            new_df.add_column(
                col_name.clone(),
                Series::new(selected, Some(col_name.clone()))?,
            )?;
        } else if let Ok(values) = df.get_column_string_values(&col_name) {
            let selected: Vec<String> = indices
                .iter()
                .filter_map(|&i| values.get(i).cloned())
                .collect();
            new_df.add_column(
                col_name.clone(),
                Series::new(selected, Some(col_name.clone()))?,
            )?;
        }
    }
    Ok(new_df)
}