如何通过CLI将CSV导入Rust Polars DataFrame？报错求助

你的代码存在三个核心问题导致执行失败：

• 标准输入被重复读取耗尽：main函数中先通过read_to_string把stdin的所有内容读到了buffer里，此时stdin数据流已经空了，再传入load_file时CsvReader无法读到任何数据。
• CsvReader未正确生成LazyFrame：CsvReader::from()返回的是配置构建器，必须调用.lazy()方法才能得到LazyFrame，直接返回构建器会触发类型不匹配错误。
• 模块导入不完整：main里用io::stdin()但未导入std::io模块，会导致编译错误。

修复后的代码 #

use std::io::stdin;
use polars::prelude::*;

fn load_file<R: std::io::Read>(reader: R) -> LazyFrame {
    CsvReader::from(reader)
        .lazy()
        .expect("Failed to parse CSV into LazyFrame")
}

fn main() {
    // 直接将标准输入传入load_file，无需提前读取buffer
    let df = load_file(stdin());
    
    // 可以添加打印验证结果
    println!("{:?}", df.collect());
}

关键改动说明 #

• 统一读取逻辑：去掉main中多余的read_to_string步骤，直接将stdin传递给CsvReader，避免数据流被耗尽。
• 正确构建LazyFrame：在CsvReader构建器后调用.lazy()方法，生成预期的LazyFrame类型。
• 泛型参数优化：将load_file的参数改为泛型R: std::io::Read，这样函数可以接受任何实现Read trait的类型（比如Stdin、Cursor<&str>等），扩展性更强。
• 简化导入：直接使用stdin()函数替代io::stdin()，避免额外的模块导入。

执行验证 #

运行原命令即可正常解析CSV：

cargo run < data/play_data.csv

内容的提问来源于stack exchange，提问作者Akmal Soliev