根据 Rust 中带极坐标的索引数据帧读取特定值

Question

我正在尝试根据 Rust 上的索引和列从数据框中读取值。我不知道如何创建索引数据框。我一直无法找到这方面的很多信息。例如，如果我的 df 是：

┌─────┬─────┬─────┬─────┬─────┐
│ ID  ┆ A   ┆ B   ┆ C   ┆ D   │
│ --- ┆ --- ┆ --- ┆ --- ┆ --- │
│ f64 ┆ f64 ┆ f64 ┆ f64 ┆ f64 │
╞═════╪═════╪═════╪═════╪═════╡
│ 0.0 ┆ 1.0 ┆ 2.0 ┆ 1.0 ┆ 1.0 │
│ 1.0 ┆ 2.0 ┆ 3.0 ┆ 1.0 ┆ 2.0 │
└─────┴─────┴─────┴─────┴─────┘

我想根据 ID 选择 A 的值。

我尝试阅读该专栏，将其转换为系列，然后转换为向量并基于索引进行读取，但一直在努力寻找工作文档。我也尝试过基于行，然后基于系列等。

感谢任何和所有帮助。

Answer 1

使用极坐标表达式：

let value = df
    .lazy()
    .filter(col("ID").eq(0.))
    .select([col("A")])
    .collect()
    .unwrap()["A"]
    .f64()
    .unwrap()
    .get(0)
    .unwrap();

还有许多其他过滤器可用。