2

我正在尝试使用 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()函数几乎可以满足我的需要，它给了我一片向量&[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)
4

1 回答 1

5

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<_>, _>>();
}

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

#[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))
}
于 2021-03-18T22:06:17.260 回答