如何通过判断列值是否存在于向量中来过滤Polars DataFrame？

云原生阿岚

2026-8-5

问题

我有一个包含“ID”列（数据类型为UInt32）的DataFrame，还有一个名为ids的向量。我希望返回仅包含“ID”列值存在于ids向量中的行的DataFrame。

最小期望示例

use polars::df;
use polars::prelude::*;

fn filter_by_id(table: &DataFrame, ids: Vec<u32>) -> DataFrame {
    df!{
        "ID" => &[1, 3, 5],
        "VALUE" => &["B", "D", "F"]
    }.unwrap()
}

fn main() {
    let table = df!{
        "ID" => &[0, 1, 2, 3, 4, 5],
        "VALUE" => &["A", "B", "C", "D", "E", "F"]
    }.unwrap();
    let ids = vec![1, 3, 5];
    let filtered_table = filter_by_id(&table, ids);
    println!("{:?}", table);
    println!("{:?}", filtered_table);
}