如何在Rust版Polars中创建固定大小向量类型的列？

Polars的df!宏默认不支持直接将固定大小数组（如[[f64;2];3]）作为列输入，因为对应的NamedFrom trait未实现。以下是几种可行的实现方式：

方法一：转换为动态列表列 #

将固定大小数组转换为Vec<Vec<f64>>，Polars原生支持这种动态列表类型，适合不需要强类型长度约束的场景：

use polars::prelude::*;

fn main() -> Result<(), PolarsError> {
    // 将固定大小数组转成Vec<Vec<f64>>
    let positions = [[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]
        .iter()
        .map(|arr| arr.to_vec())
        .collect::<Vec<_>>();

    let df = df!(
        "a" => &[1, 2, 3],
        "b" => positions
    )?;

    println!("{}", df);
    Ok(())
}

方法二：自定义强类型结构体列 #

如果需要严格保证向量的维度（比如强制2D点、3D坐标），可以定义自定义结构体并实现IntoSeries trait，构建强类型的结构体列：

use polars::prelude::*;
use polars::export::arrow::array::StructArray;
use polars::export::arrow::datatypes::{Field, DataType};

#[derive(Debug, Clone)]
struct Point2D {
    x: f64,
    y: f64,
}

impl Point2D {
    fn new(x: f64, y: f64) -> Self {
        Self { x, y }
    }
}

// 实现IntoSeries，将Point2D的Vec转为Polars支持的结构体Series
impl IntoSeries for Vec<Point2D> {
    fn into_series(self) -> Series {
        let x_values: Vec<f64> = self.iter().map(|p| p.x).collect();
        let y_values: Vec<f64> = self.iter().map(|p| p.y).collect();

        let x_series = Series::new("x", x_values);
        let y_series = Series::new("y", y_values);

        let struct_array = StructArray::from(vec![x_series.into_arrow(), y_series.into_arrow()]);
        let field = Field::new("point", DataType::Struct(vec![
            Field::new("x", DataType::Float64, false),
            Field::new("y", DataType::Float64, false),
        ]), false);

        Series::from_arrow(field.name(), struct_array.into()).unwrap()
    }
}

fn main() -> Result<(), PolarsError> {
    let points = vec![
        Point2D::new(1.0, 2.0),
        Point2D::new(3.0, 4.0),
        Point2D::new(5.0, 6.0),
    ];

    let df = df!(
        "a" => &[1, 2, 3],
        "b" => points
    )?;

    println!("{}", df);
    Ok(())
}

方法三：手动构建Series #

绕过df!宏，直接用Series::new构建列表列，步骤更清晰：

use polars::prelude::*;

fn main() -> Result<(), PolarsError> {
    let a_series = Series::new("a", &[1, 2, 3]);
    // 直接将固定数组转换为Vec<Vec<f64>>后构建Series
    let b_series = Series::new(
        "b",
        &[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]
            .iter()
            .map(|arr| arr.to_vec())
            .collect::<Vec<_>>()
    );

    let df = DataFrame::new(vec![a_series, b_series])?;
    println!("{}", df);
    Ok(())
}

原代码报错原因 #

df!宏要求输入的序列类型实现NamedFrom trait，而Polars并未为固定大小数组（如[[f64;2];3]）实现该 trait，因此需要转换为Polars支持的类型（如动态列表、自定义结构体）才能创建DataFrame。

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