使用 PyO3 将列表列表作为参数从 Python 传递给 Rust

Question

我正在尝试使用 Py03 将列表列表从 Python 传递给 Rust。我试图将其传递给的函数具有此签名：

pub fn k_nearest_neighbours(k: usize, x: &[[f32; 2]], y: &[[f32; 3]]) -> Vec<Option<f32>>

我正在为预先存在的库编写绑定，因此我无法更改原始代码。我目前的做事方式是这样的：

// This is example code == DOES NOT WORK
#[pyfunction] // make a new function within a new library with pyfunction macro
fn k_nearest_neighbours(k: usize, x: Vec<Vec<f32>>, y: Vec<f32>) -> Vec<Option<f32>> {
    // reformat input where necessary
    let x_slice = x.as_slice();
    // return original lib's function return
    classification::k_nearest_neighbours(k, x_slice, y)
}

x.as_slice() 函数 almost 做我需要的，它给了我一片向量 &[Vec<f32>]，而不是一片片 &[[f32; 3]].

我希望能够运行这个Python代码：

from rust_code import k_nearest_neighbours as knn  # this is the Rust function compiled with PyO3

X = [[0.0, 1.0], [2.0, 3.0], [4.0, 5.0], [0.06, 7.0]]
train = [
        [0.0, 0.0, 0.0],
        [0.5, 0.5, 0.0],
        [3.0, 3.0, 1.0],
        [4.0, 3.0, 1.0],
    ]

k = 2
y_true = [0, 1, 1, 1]
y_test = knn(k, X, train)
assert(y_true == y_test)

Answer 1

查看 k_nearest_neighbours 的签名表明它期望 [f32; 2] 和 [f32; 3] 是数组，而不是切片（例如 &[f32].

数组在编译时具有静态已知大小，而切片是动态大小的。向量也是如此，您无法控制示例中内部向量的长度。因此，您最终会得到从输入向量到预期数组的错误转换。

您可以使用 TryFrom 从切片转换为数组，即：

use std::convert::TryFrom;
fn main() {
    let x = vec![vec![3.5, 3.4, 3.6]];
    let x: Result<Vec<[f32; 3]>, _> = x.into_iter().map(TryFrom::try_from).collect::<Result<Vec<_>, _>>();
}

综上所述，您的函数将需要 return 在输入不正确时出错，并且您将需要创建一个包含可传递给函数的数组的新向量：

#[pyfunction] // make a new function within a new library with pyfunction macro
fn k_nearest_neighbours(k: usize, x: Vec<Vec<f32>>, y: Vec<f32>) -> PyResult<Vec<Option<f32>>> {
    let x = x.into_iter().map(TryFrom::try_from).collect::<Result<Vec<_>, _>>();
    let y = y.into_iter().map(TryFrom::try_from).collect::<Result<Vec<_>, _>>();
    // Error handling is missing here, you'll need to look into PyO3's documentation for that
    ...
    // return original lib's function return
    Ok(classification::k_nearest_neighbours(k, &x, &y))
}

使用 PyO3 将列表列表作为参数从 Python 传递给 Rust

Pass list of lists as argument to Rust from Python using PyO3

python

rust

pyo3