API documentation

Requirements

Example

Write a Python module in Rust

[lib]name ="rust_ext"crate-type = ["cdylib"][dependencies]pyo3 = {version ="0.27" }numpy ="0.27"

#[pyo3::pymodule]mod rust_ext{use numpy::ndarray::{ArrayD,ArrayViewD,ArrayViewMutD};use numpy::{IntoPyArray,PyArrayDyn,PyReadonlyArrayDyn,PyArrayMethods};use pyo3::{pyfunction,PyResult,Python,Bound};// example using immutable borrows producing a new arrayfnaxpy(a:f64,x:ArrayViewD<'_,f64>,y:ArrayViewD<'_,f64>) ->ArrayD<f64>{        a*&x +&y}// example using a mutable borrow to modify an array in-placefnmult(a:f64,mutx:ArrayViewMutD<'_,f64>){        x *= a;}// wrapper of `axpy`#[pyfunction(name ="axpy")]fnaxpy_py<'py>(py:Python<'py>,a:f64,x:PyReadonlyArrayDyn<'py,f64>,y:PyReadonlyArrayDyn<'py,f64>,) ->Bound<'py,PyArrayDyn<f64>>{let x = x.as_array();let y = y.as_array();let z =axpy(a, x, y);        z.into_pyarray(py)}// wrapper of `mult`#[pyfunction(name ="mult")]fnmult_py<'py>(a:f64,x:&Bound<'py,PyArrayDyn<f64>>){let x =unsafe{ x.as_array_mut()};mult(a, x);}}

Execute a Python program from Rust and get results

[package]name ="numpy-test"[dependencies]pyo3 = {version ="0.27",features = ["auto-initialize"] }numpy ="0.27"

use numpy::{PyArray1,PyArrayMethods};use pyo3::{types::{IntoPyDict,PyAnyMethods},PyResult,Python, ffi::c_str};fnmain() ->PyResult<()>{Python::attach(|py|{let np = py.import("numpy")?;let locals =[("np", np)].into_py_dict(py)?;let pyarray = py.eval(c_str!("np.absolute(np.array([-1, -2, -3], dtype='int32'))"),Some(&locals),None)?.cast_into::<PyArray1<i32>>()?;let readonly = pyarray.readonly();let slice = readonly.as_slice()?;assert_eq!(slice,&[1,2,3]);Ok(())})}

Dependency on ndarray

numpy ="0.27"ndarray ="0.15"

cargo update --package ndarray:0.16.1 --precise 0.15.6

Contributing