如何将Rust Polars的list[list[str]]列转换为list[Foo]类型列？

这种转换是可行的，但需要注意两个关键细节：

• Polars的ObjectType要求自定义类型拥有所有权，不能使用引用类型（如&str），否则无法安全管理生命周期；
• 自定义类型必须实现Polars的ObjectTrait，才能被识别为合法的Object类型。

步骤1：调整Foo结构体 #

将&str替换为String，确保类型拥有所有权：

#[derive(Debug, Clone)]
struct Foo {
    f1: String,
    f2: String,
}

步骤2：为Foo实现ObjectTrait #

这是Polars识别自定义类型的必要条件：

use polars::prelude::*;

impl ObjectTrait for Foo {
    fn type_name() -> &'static str {
        "Foo"
    }
}

步骤3：执行转换逻辑 #

通过apply方法遍历原列的每个嵌套列表，将内部的[str, str]转换为Foo实例，最终生成list[Foo]类型的Series：

fn main() -> PolarsResult<()> {
    // 构造测试数据
    let data = vec![
        vec![vec!["a", "b"], vec!["c", "d"], vec!["e", "f"]],
        vec![vec!["g", "h"], vec!["i", "j"], vec!["k", "l"]],
        vec![vec!["m", "n"], vec!["o", "p"], vec!["q", "r"]],
    ];
    let df = DataFrame::new(vec![Series::new("Foos", data)])?;

    // 转换列：list[list[str]] -> list[Foo]
    let foos_series = df
        .column("Foos")?
        .list()?
        .apply(|opt_inner_list| {
            let Some(inner_list) = opt_inner_list else {
                return Ok(None);
            };
            let inner_list = inner_list.list()?;
            let foo_list: PolarsResult<Vec<Foo>> = inner_list
                .into_iter()
                .map(|opt_pair| {
                    let Some(pair) = opt_pair else {
                        return Err(PolarsError::ValueError("缺失元素对".into()));
                    };
                    let pair = pair.str()?;
                    if pair.len() != 2 {
                        return Err(PolarsError::ValueError("元素对必须包含恰好两个元素".into()));
                    }
                    let f1 = pair.get(0)?.unwrap().to_string();
                    let f2 = pair.get(1)?.unwrap().to_string();
                    Ok(Foo { f1, f2 })
                })
                .collect();
            Ok(foo_list?.into())
        })?
        .into_series();

    // 验证结果
    let new_df = df.with_column(foos_series.rename("Foos_Foo"))?;
    println!("{new_df}");

    Ok(())
}

对之前尝试的说明 #

• ChunkedArray<ObjectType<T>>：失败原因是原Foo使用&str引用类型，无法满足ObjectTrait的生命周期要求，且未实现ObjectTrait；
• StructArray：StructArray用于表示结构体列（每一行是一个独立结构体），而你需要的是列表元素为结构体的列，两者适用场景不同，因此不匹配。

内容的提问来源于stack exchange，提问作者Neotenic Primate