如何创建一个PyArray来提供给方法

Question

我有一个方法foo，签名如下:

pub fn foo(data: PyReadonlyArrayDyn<f64>) {
...
}

在我的测试中，我试图创建一个测试数组来输入foo

pyo3::Python::with_gil(|py| {
let vec = vec![1.0, 2.0, 3.0, 4.0];
py_array = PyArray::from_vec(py, vec);
readonly = py_array.readonly();
foo(readonly);
}

不幸的是，rust告诉我:

mismatched types
expected struct `PyReadonlyArray<'_, f64, Dim<IxDynImpl>>`
found struct `PyReadonlyArray<'_, {float}, Dim<[usize; 1]>>`

显然，我遇到了一个关于维度的问题。我必须如何创建PyReadonlyArray才能将其馈入foo?

Answer 1

CallPyArray::to_dyn():

pyo3::Python::with_gil(|py| {
let vec = vec![1.0, 2.0, 3.0, 4.0];
py_array = PyArray::from_vec(py, vec);
readonly = py_array.to_dyn().readonly();
foo(readonly);
}