Rust读写CSV文件 一维Vec类型元素、二维Vec类型元素写入CSV文件

本文主要介绍Rust读写CSV文件方法, Vec类型元素基本操作方法,Rust把一维Vec类型元素、二维Vec类型元素写入CSV文件方法。

实例测试:

要求读"log.csv"文件数据,把"时间"列数据和"次数"列数据写入日志处理结果1.csv文件,把"时间"列数据写入日志处理结果2.csv文件。

log.csv文件:

日志处理结果1.csv文件:

日志处理结果2.csv文件:

实例代码

Cargo.toml添加依赖

rust 复制代码
[dependencies]
csv = "1.3.0"

main.rc文件代码

rust 复制代码
#[macro_use]
pub mod macros;
use csv::ReaderBuilder;
use csv::WriterBuilder;
use std::fs::File;
use std::str;


fn main() -> Result<(), Box<dyn std::error::Error>> {
    //创建一维Vec
    let vec_headers = Vec::from([
        "日期时间".to_string(),
        "次数".to_string(),
    ]); 

    //创建一维Vec
    let vec_headers_2 = Vec::from([
        "日期时间".to_string(),
    ]); 
    
    // 创建二维Vec
    let mut vec_matrix = Vec::new();
    //创建一维Vec
    let mut vec_date = Vec::new();  

    let input_path = "log.csv";

    /* 读取csv日志文件,并提取主要数据 */
    read_csv_file(&input_path, &mut vec_matrix, &mut vec_date).expect("Failed to read CSV file");

    /* 处理结果写入csv文件 */
    let out_path_1 = "日志处理结果1.csv";
    let _z = write_csv_file(&out_path_1, &vec_headers, &vec_matrix);
    let out_path_2 = "日志处理结果2.csv";
    let _z = write_csv_file_2(&out_path_2, &vec_headers_2,&vec_date);

    //打印等待用户按回车键,退出控制台
    println!("温馨提示:请按回车键退出......");

    // 等待用户按下回车键
    let _ = std::io::stdin().read_line(&mut String::new()).expect("TODO:panic message");
 
    Ok(())
}


/*读取csv文件,并提取主要数据 */
pub fn read_csv_file(
    path: &str,
    vec_matrix: &mut Vec<Vec<String>>,
    vec_date: &mut Vec<String>,
) -> Result<(), Box<dyn std::error::Error>> {
    let file = File::open(path)?;
    let mut reader = ReaderBuilder::new().from_reader(file);

    for result in reader.records() {
        //创建一维Vec
        let mut vec = Vec::new();

        let record = result?;
        // 处理每一行的数据
        //println!("{:?}", record);
        //println!("{:?}", record.get(0));
        //println!("{}", record.get(0).unwrap());
        vec_date.push(record.get(0).unwrap().to_string());

        vec.push(record.get(0).unwrap().to_string());
        vec.push(record.get(2).unwrap().to_string());

        //打印
        //println!("{:?}", vec);

        //添加到二维Vec
        vec_matrix.push(vec);
    }

    //打印
    println!("vec_date:");
    println!("{:?}", vec_date);
    //打印
    println!("vec_matrix包含记录{}条:", vec_matrix.len());
    for vec_item in vec_matrix.iter() {
        println!("{:?}", vec_item);
    }

    Ok(())
}

/*把二维Vec写入csv文件 */
pub fn write_csv_file(
    path: &str,
    headers: &Vec<String>,
    records: &Vec<Vec<String>>,
) -> Result<(), Box<dyn std::error::Error>> {
    let file = File::create(path)?;
    let mut writer = WriterBuilder::new().from_writer(file);

    writer.write_record(headers)?;

    for record in records {
        writer.write_record(record)?;
    }
    // 确保所有数据都被写入
    writer.flush()?;

    println!("已成功写入"{}"文件", path);

    Ok(())
}


/*把一维Vec写入csv文件 */
pub fn write_csv_file_2(
    path: &str,
    headers: &Vec<String>,
    record: &Vec<String>,
) -> Result<(), Box<dyn std::error::Error>> {
    let file = File::create(path)?;
    let mut writer = WriterBuilder::new().from_writer(file);

    writer.write_record(headers)?;

    // 遍历Vec,将每个元素作为一行写入CSV
    for item in record {
        // 将每个String元素作为单个记录写入CSV
        writer.write_record(&[&item])?;
    }
    // 确保所有数据都被写入
    writer.flush()?;

    println!("已成功写入"{}"文件", path);

    Ok(())
}

macros.rs文件代码

rust 复制代码
#[allow(unused_macros)]
#[macro_export]
macro_rules! hashmap {
    ($( $key: expr => $val: expr ),*) => {{
         let mut map = ::std::collections::HashMap::new();
         $( map.insert($key, $val); )*
         map
    }}
}

#[allow(unused_macros)]
#[macro_export]
macro_rules! hashmap_s {
    ($( $key: expr => $val: expr ),*) => {{
         let mut map = ::std::collections::HashMap::new();
         $( map.insert($key.to_string(), $val); )*
         map
    }}
}

#[macro_export]
macro_rules! vec_s {
    ($( $val: expr ),*) => {{
        let mut vec = ::std::vec::Vec::new();
        $( vec.push($val.to_string()); )*
        vec
    }}
}

#[macro_export]
macro_rules! hashset {
    ($( $val: expr ),*) => {{
        let mut vec = ::std::collections::HashSet::new();
        $( vec.insert($val); )*
        vec
    }}
}

#[macro_export]
macro_rules! hashset_s {
    ($( $val: expr ),*) => {{
        let mut vec =  ::std::collections::HashSet::new();
        $( vec.insert($val.to_string()); )*
        vec
    }}
}
运行结果

相关推荐
hccee12 分钟前
C# IO文件操作
开发语言·c#
hummhumm16 分钟前
第 25 章 - Golang 项目结构
java·开发语言·前端·后端·python·elasticsearch·golang
J老熊26 分钟前
JavaFX:简介、使用场景、常见问题及对比其他框架分析
java·开发语言·后端·面试·系统架构·软件工程
zmd-zk41 分钟前
flink学习(2)——wordcount案例
大数据·开发语言·学习·flink
好奇的菜鸟1 小时前
Go语言中的引用类型:指针与传递机制
开发语言·后端·golang
Alive~o.01 小时前
Go语言进阶&依赖管理
开发语言·后端·golang
花海少爷1 小时前
第十章 JavaScript的应用课后习题
开发语言·javascript·ecmascript
手握风云-1 小时前
数据结构(Java版)第二期:包装类和泛型
java·开发语言·数据结构
喵叔哟1 小时前
重构代码中引入外部方法和引入本地扩展的区别
java·开发语言·重构
尘浮生1 小时前
Java项目实战II基于微信小程序的电影院买票选座系统(开发文档+数据库+源码)
java·开发语言·数据库·微信小程序·小程序·maven·intellij-idea