如何从Polars DataFrame中提取结构体的Vec集合？

你可以通过Polars提供的行遍历API，将DataFrame中的数据逐行转换为Person结构体，替代手动定义的rows向量。以下是具体实现：

基础实现（按列索引取值） #

use polars::prelude::*;

pub struct Person {
    id: u32,
    name: String,
    age: u32,
}

// 假设df已按你的代码初始化完成
let df = df!(
    "id" => &[1,2,3],
    "name" => &["John", "Jane", "Bobby"],
    "age" => &[32, 28, 45]
).unwrap();

// 转换逻辑
let rows: Result<Vec<Person>, PolarsError> = df.iter_rows()
    .map(|row| {
        // 按列索引提取并转换数据类型
        let id = row.get(0)?.try_extract()?;
        let name = row.get(1)?.try_extract()?;
        let age = row.get(2)?.try_extract()?;
        Ok(Person { id, name, age })
    })
    .collect();

// 处理转换结果（实际项目中建议优雅处理错误，避免unwrap）
let rows = rows.unwrap();

更安全的实现（按列名取值） #

如果担心列顺序变化导致索引错误，可以直接通过列名取值：

let rows: Result<Vec<Person>, PolarsError> = df.iter_rows()
    .map(|row| {
        let id = row.get("id")?.try_extract()?;
        let name = row.get("name")?.try_extract()?;
        let age = row.get("age")?.try_extract()?;
        Ok(Person { id, name, age })
    })
    .collect();

Wasm环境注意事项 #

1. 确保使用的Polars Wasm版本支持iter_rows API（当前polars-wasm已支持该功能）
2. 避免直接使用unwrap处理错误，建议通过match或?传递错误，防止Wasm模块崩溃
3. 转换后的Vec<Person>可直接传入leptos-struct-table使用，符合组件对数据源的要求

内容的提问来源于stack exchange，提问作者Dean MacGregor