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
    }}
}
运行结果

相关推荐
秃头佛爷40 分钟前
Python学习大纲总结及注意事项
开发语言·python·学习
待磨的钝刨41 分钟前
【格式化查看JSON文件】coco的json文件内容都在一行如何按照json格式查看
开发语言·javascript·json
XiaoLeisj3 小时前
【JavaEE初阶 — 多线程】单例模式 & 指令重排序问题
java·开发语言·java-ee
励志成为嵌入式工程师4 小时前
c语言简单编程练习9
c语言·开发语言·算法·vim
捕鲸叉4 小时前
创建线程时传递参数给线程
开发语言·c++·算法
A charmer4 小时前
【C++】vector 类深度解析:探索动态数组的奥秘
开发语言·c++·算法
Peter_chq4 小时前
【操作系统】基于环形队列的生产消费模型
linux·c语言·开发语言·c++·后端
记录成长java6 小时前
ServletContext,Cookie,HttpSession的使用
java·开发语言·servlet
前端青山6 小时前
Node.js-增强 API 安全性和性能优化
开发语言·前端·javascript·性能优化·前端框架·node.js
hikktn6 小时前
如何在 Rust 中实现内存安全:与 C/C++ 的对比分析
c语言·安全·rust