求助：在Rust中使用Plotters绘制Polars DataFrame数据

我想在Rust里用Plotters库把Polars DataFrame的数据绘制成图像，参考相关教程写了代码，但没法实现预期的多列绘图逻辑，目前仅能完成单列数据的绘制。

尝试获取迭代器的代码 #

let iters = dataset
    .columns(["x", "y"])?
    .iter()
    .map(|s| Ok(s.f64()?.into_iter()))
    .collect::<Result<Vec<_>, Box<dyn Error>>>()?;

期望实现的绘图逻辑 #

chart
    .draw_series(
        x.zip(y).map(|(x, y)| Circle::new((x, x), 3, BLUE.filled())),
    )?
    .label("data")
    .legend(|(x, y)| Circle::new((x, y), 3, BLUE.filled()));

当前仅能运行的单列绘制代码 #

let mut iters = dataset.groundtruth
    .columns(["x", "y"])?
    .iter()
    .map(|s| Ok(s.f64()?.into_iter()))
    .collect::<Result<Vec<_>, Box<dyn Error>>>()?;

let x = &mut iters[0];

chart
    .draw_series(
        // x.filter(|x| x.is_some())
        x.map(|(x)| Circle::new((x.unwrap(), x.unwrap()), 3, BLUE.filled())),
    )?
    .label("data")
    .legend(|(x, y)| Circle::new((x, y), 3, BLUE.filled()));

我怀疑是生命周期问题导致的，希望能解决Polars和Plotters结合绘制多列数据的问题。

解决方案 #

问题核心在于列迭代器的所有权处理和空值兼容：columns返回的引用迭代器无法让你同时获取两列的所有权进行zip迭代，且Plotters要求迭代器提供拥有所有权的绘图元素，同时还要处理Polars列中可能存在的空值。

修改后的完整可运行代码

use polars::prelude::*;
use plotters::prelude::*;
use std::error::Error;

fn main() -> Result<(), Box<dyn Error>> {
    // 替换为你的DataFrame实例
    let dataset = DataFrame::new(vec![
        Series::new("x", vec![1.0, 2.0, 3.0, 4.0, 5.0]),
        Series::new("y", vec![2.0, 4.0, 5.0, 7.0, 8.0]),
    ])?;

    // 直接获取x、y列的所有权，转为f64迭代器
    let x_iter = dataset.column("x")?.f64()?.into_iter();
    let y_iter = dataset.column("y")?.f64()?.into_iter();

    // 初始化绘图上下文
    let root = BitMapBackend::new("polars_plot.png", (640, 480)).into_drawing_area();
    root.fill(&WHITE)?;

    let mut chart = ChartBuilder::on(&root)
        .caption("Polars + Plotters 数据点绘图", ("sans-serif", 20))
        .x_label_area_size(40)
        .y_label_area_size(40)
        .build_cartesian_2d(0.0..6.0, 0.0..9.0)?;

    chart.configure_mesh().draw()?;

    // 合并迭代器、过滤空值、生成绘图元素
    chart.draw_series(
        x_iter.zip(y_iter)
            .filter(|(x, y)| x.is_some() && y.is_some())
            .map(|(x, y)| Circle::new((x.unwrap(), y.unwrap()), 3, BLUE.filled()))
    )?
    .label("数据点")
    .legend(|(x, y)| Circle::new((x, y), 3, BLUE.filled()));

    chart.configure_series_labels().draw()?;
    root.present()?;

    Ok(())
}

关键细节说明

• 所有权处理：直接通过dataset.column("x")获取列所有权，再调用into_iter()生成拥有所有权的迭代器，避免引用导致的生命周期冲突
• 空值兼容：用filter过滤掉任意一列的空值，防止unwrap()触发panic（若确定数据无空值可省略，但保留更鲁棒）
• 迭代器适配：zip后的迭代器输出(Option<f64>, Option<f64>)，处理后转换为Plotters支持的坐标格式，满足draw_series的迭代器要求

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