Crate ndarray_csv[][src]

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::{read, write};
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);
        write(&array, &mut writer).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 = read((2, 3), &mut reader).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

ReadError

Functions

read

Read CSV data into a new ndarray with the given shape

write

Write this ndarray into CSV format