mirror of
https://github.com/Magnus167/rustframe.git
synced 2025-08-20 13:00:00 +00:00
Add DataFrame usage examples to README.md
This commit is contained in:
parent
2607d9c3b0
commit
58acea8467
98
README.md
98
README.md
@ -98,3 +98,101 @@ assert!(check);
|
|||||||
|
|
||||||
|
|
||||||
```
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## DataFrame Usage Example
|
||||||
|
|
||||||
|
```rust
|
||||||
|
use rustframe::{
|
||||||
|
dataframe::{DataFrame, TypedFrame, DataFrameColumn},
|
||||||
|
frame::{Frame, RowIndex},
|
||||||
|
matrix::Matrix,
|
||||||
|
};
|
||||||
|
|
||||||
|
// Helper to create a simple f64 TypedFrame (similar to test helpers)
|
||||||
|
fn create_f64_typed_frame(name: &str, data: Vec<f64>, index: Option<RowIndex>) -> TypedFrame {
|
||||||
|
let rows = data.len();
|
||||||
|
let matrix = Matrix::from_cols(vec![data]);
|
||||||
|
let frame_index = index.unwrap_or(RowIndex::Range(0..rows));
|
||||||
|
TypedFrame::F64(Frame::new(
|
||||||
|
matrix,
|
||||||
|
vec![name.to_string()],
|
||||||
|
Some(frame_index),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper to create a simple i64 TypedFrame
|
||||||
|
fn create_i64_typed_frame(name: &str, data: Vec<i64>, index: Option<RowIndex>) -> TypedFrame {
|
||||||
|
let rows = data.len();
|
||||||
|
let matrix = Matrix::from_cols(vec![data]);
|
||||||
|
let frame_index = index.unwrap_or(RowIndex::Range(0..rows));
|
||||||
|
TypedFrame::I64(Frame::new(
|
||||||
|
matrix,
|
||||||
|
vec![name.to_string()],
|
||||||
|
Some(frame_index),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper to create a simple String TypedFrame
|
||||||
|
fn create_string_typed_frame(
|
||||||
|
name: &str,
|
||||||
|
data: Vec<String>,
|
||||||
|
index: Option<RowIndex>,
|
||||||
|
) -> TypedFrame {
|
||||||
|
let rows = data.len();
|
||||||
|
let matrix = Matrix::from_cols(vec![data]);
|
||||||
|
let frame_index = index.unwrap_or(RowIndex::Range(0..rows));
|
||||||
|
TypedFrame::String(Frame::new(
|
||||||
|
matrix,
|
||||||
|
vec![name.to_string()],
|
||||||
|
Some(frame_index),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn main() {
|
||||||
|
// 1. Create a DataFrame with different data types
|
||||||
|
let col_a = create_f64_typed_frame("A", vec![1.0, 2.0, 3.0], None);
|
||||||
|
let col_b = create_i64_typed_frame("B", vec![10, 20, 30], None);
|
||||||
|
let col_c = create_string_typed_frame(
|
||||||
|
"C",
|
||||||
|
vec!["apple".to_string(), "banana".to_string(), "cherry".to_string()],
|
||||||
|
None,
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut df = DataFrame::new(
|
||||||
|
vec![col_a, col_b, col_c],
|
||||||
|
vec!["A".to_string(), "B".to_string(), "C".to_string()],
|
||||||
|
None,
|
||||||
|
);
|
||||||
|
|
||||||
|
println!("Initial DataFrame:\n{:?}", df);
|
||||||
|
println!("Columns: {:?}", df.columns());
|
||||||
|
println!("Rows: {}", df.rows());
|
||||||
|
|
||||||
|
// 2. Accessing columns
|
||||||
|
if let DataFrameColumn::F64(col_a_data) = df.column("A") {
|
||||||
|
println!("Column 'A' (f64): {:?}", col_a_data);
|
||||||
|
}
|
||||||
|
|
||||||
|
if let DataFrameColumn::String(col_c_data) = df.column("C") {
|
||||||
|
println!("Column 'C' (String): {:?}", col_c_data);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. Add a new column
|
||||||
|
let new_col_d = create_f64_typed_frame("D", vec![100.0, 200.0, 300.0], None);
|
||||||
|
df.add_column("D".to_string(), new_col_d);
|
||||||
|
println!("\nDataFrame after adding column 'D':\n{:?}", df);
|
||||||
|
println!("Columns after add: {:?}", df.columns());
|
||||||
|
|
||||||
|
// 4. Rename a column
|
||||||
|
df.rename_column("A", "Alpha".to_string());
|
||||||
|
println!("\nDataFrame after renaming 'A' to 'Alpha':\n{:?}", df);
|
||||||
|
println!("Columns after rename: {:?}", df.columns());
|
||||||
|
|
||||||
|
// 5. Delete a column
|
||||||
|
let _deleted_col_b = df.delete_column("B");
|
||||||
|
println!("\nDataFrame after deleting column 'B':\n{:?}", df);
|
||||||
|
println!("Columns after delete: {:?}", df.columns());
|
||||||
|
}
|
||||||
|
```
|
||||||
|
Loading…
x
Reference in New Issue
Block a user