如何在Rust中从Polars Series提取数值？

在使用Rust的Polars库时，能够正常构建、排序、查询DataFrame，但从Series中获取到的是AnyValue类型对象（例如输出中的Int32(50)），无法直接参与数值运算。尝试执行let y = x - src_start;时，出现如下编译错误：

error[E0277]: cannot subtract `AnyValue<'_>` from `{integer}`
   --> src/main.rs:165:19
    |
165 |         let y = x - src_start;
    |                   ^ no implementation for `{integer} - AnyValue<'_>`
    |
    = help: the trait `Sub<AnyValue<'_>>` is not implemented for `{integer}`
    = help: the following other types implement trait `Sub<Rhs>`:
              <isize as Sub>
              <isize as Sub<&isize>>
              <i8 as Sub>
              <i8 as Sub<&i8>>
              <i16 as Sub>
              <i16 as Sub<&i16>>
              <i32 as Sub>
              <i32 as Sub<&i32>>
            and 56 others

For more information about this error, try `rustc --explain E0277`.

用户的示例代码：

let s = Series::new("source", &[0, 98, 50]);
let d: Series = Series::new("destination", &[0, 50, 52]);
let r: Series = Series::new("range", &[-1, 2, 48]);
let df = DataFrame::new(vec![s, d, r]).unwrap();
let sdf = df.sort(["source", "destination", "range"], false, true).unwrap();

let x = 79;
// find row where seed is between source and source+range
let sources = sdf.column("source").unwrap();
// find index of value less than seed
let mask = sources.lt_eq(x).unwrap();
let less_df = sdf.filter(&mask).unwrap();

let sources = less_df.column("source").unwrap();
let destinations = less_df.column("destination").unwrap();
let ranges = less_df.column("range").unwrap();

let num_sources = sources.len();
let src_start = sources.get(num_sources - 1).unwrap();
let dst_start = destinations.get(num_sources - 1).unwrap();
let range = ranges.get(num_sources - 1).unwrap();
println!("{:?} {:?} {:?}", src_start, dst_start, range);

运行输出：

Int32(50) Int32(52) Int32(48)

解决方案 #

方法1：模式匹配提取数值 #

AnyValue是Polars定义的枚举类型，包含了所有可能的数据类型变体。可以通过模式匹配精准提取对应类型的数值：

let src_start = sources.get(num_sources - 1).unwrap();
// 提取i32类型的数值
let src_start_val = match src_start {
    AnyValue::Int32(val) => val,
    _ => panic!("source列的类型不是预期的Int32"),
};
let y = x - src_start_val;

方法2：使用as_*方法直接转换 #

如果已经确定列的类型为Int32，可以使用AnyValue提供的as_i32()方法直接转换（类型不匹配时会返回None，需要处理）：

let src_start_val = src_start.as_i32().unwrap();
let y = x - src_start_val;

方法3：批量转换为具体类型数组 #

如果需要对整列数据进行操作，建议先将Series转换为对应类型的数组，后续操作更高效：

// 将source列转换为i32数组（处理可能的缺失值）
let sources_arr: Vec<i32> = less_df.column("source")
    .unwrap()
    .i32() // 转换为Int32Chunked
    .unwrap()
    .into_iter()
    .flatten() // 过滤掉缺失值（None）
    .collect();

let src_start_val = sources_arr[num_sources - 1];
let y = x - src_start_val;

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