Crate ndarray_csv
source ·Expand description
Easily read and write homogeneous CSV data to and from 2D ndarrays.
extern crate csv;
extern crate ndarray;
extern crate ndarray_csv;
use csv::{ReaderBuilder, WriterBuilder};
use ndarray::Array;
use ndarray_csv::{Array2Reader, Array2Writer};
use std::fs::File;
fn main() {
// Our 2x3 test array
let array = Array::from_vec(vec![1, 2, 3, 4, 5, 6]).into_shape((2, 3)).unwrap();
// Write the array into the file.
{
let file = File::create("test.csv").expect("creating file failed");
let mut writer = WriterBuilder::new().has_headers(false).from_writer(file);
writer.serialize_array2(&array).expect("write failed");
}
// Read an array back from the file
let file = File::open("test.csv").expect("opening file failed");
let mut reader = ReaderBuilder::new().has_headers(false).from_reader(file);
let array_read = reader.deserialize_array2((2, 3)).expect("read failed");
// Ensure that we got the original array back
assert_eq!(array_read, array);
}
This project uses cargo-make for builds; to build,
run cargo make all
.
To prevent denial-of-service attacks, do not read in untrusted CSV streams of unbounded length;
this can be implemented with std::io::Read::take
.
Enums
Traits
An extension trait; this is implemented by
&mut csv::Reader
An extension trait; this is implemented by
&mut csv::Writer