如何在 Rust 中迭代 pyo3 PyObject？

Question

我有一个预导入的模块，我正在使用 python gil 调用一个方法，如下所示。

Python::with_gil(|py| {
    let res = module.call_method1(py, "my_method", (arg1, arg2))?;
})

这会返回 rust 对象 PyObject，但是返回的是一个 python list。我想迭代这个列表，将内部转换为我可以在 Rust 中使用的东西（它是 Numpy 数组的 python 列表，我正在使用 numpy/ndarray 板条箱）。

我对如何迭代这个有点困惑。如果我尝试将 cast_as 转换为 PyList，我会收到警告：UnsafeCell无法在线程之间安全地共享。看来extract也不起作用。

如何迭代这个 PyObject？谢谢。

编辑：根据要求添加更多详细信息

如果您使用的是 python 打字系统，则 python 返回的值是一个 List[numpy.ndarray]。由于每个 numpy 数组的长度可能不同，我不能将其全部转换为 python 中的 numpy 数组并传递它。示例输出如下：

[array([214.17725372, 192.78236675, 354.27965546, 389.84558392,
          0.99999297])]

我在 Rust 中尝试过的内容：

• let pylist = res.cast_as::(py)?;

  无法编译：UnsafeCell;无法在线程之间安全地共享。

• let pylist = res.extract::(py)?;

  无法编译：未针对“PyList”实现特征“PyClass”。请注意，我在顶部有use pyo3::prelude::*;。

• let pyany = res.extract::>>(py)?;

  无法编译：特征绑定 'Vec>: pyo3::FromPyObject<'_>'不满意。这个 PyArray 来自 numpy 板条箱。

Answer 1

我看到您返回了带有 dtype=float 的 numpy 数组列表。如果您愿意使用其他依赖项，可以使用 rust-numpy 来映射 numpy 数组到 ndarrays。

ndarrays 有一个 .to_vec() 方法来转换为标准 Rust 容器。

在您的情况下，您可以执行以下操作（未经测试）：

use numpy::PyArray1;
use pyo3::{types::IntoPyDict, PyResult, Python};

Python::with_gil(|py| {
    let res: Vec<&PyArray1<f32>> = module // <-- Notice the return type
                                 .call_method1(py, "my_method", (arg1, arg2))?
                                 .extract()?; // <--  Notice the extract
})

可以找到 Python 和 Rust 类型之间的映射

println!("I just got a numpy array from python {:?}", res[0]);
println!("I converted that to a vector here: {:?}", res[0].to_vec());