rust-numpy

Rust binding of NumPy C-API

API documentation

• Latest release(possibly broken)
• Current Master

Requirements

• current nightly rust (see PyO3/pyo3#5 for nightly features, and https://github.com/PyO3/pyo3/blob/master/build.rs for minimum required version)
• some rust libraries
  • ndarray for rust-side matrix library
  • pyo3 for cpython binding
  • and more (see Cargo.toml)
• numpy installed in your python environments(e.g. via pip install numpy)

Note From 0.3, we migrated from rust-cpython to pyo3. If you want to use rust-cpython, use version 0.2.1 from crates.io.

Example

Exec python program and get data

[package]
name = "numpy-test"

[dependencies]
pyo3 = "^0.4.1"
numpy = "0.3"

extern crate numpy;
extern crate pyo3;
use numpy::{IntoPyResult, PyArray, PyArrayModule};
use pyo3::prelude::{ObjectProtocol, PyDict, PyResult, Python};

fn main() -> Result<(), ()> {
    let gil = Python::acquire_gil();
    main_(gil.python()).map_err(|e| {
        eprintln!("error! :{:?}", e);
        // we can't display python error type via ::std::fmt::Display
        // so print error here manually
        e.print_and_set_sys_last_vars(gil.python());
    })
}

fn main_<'py>(py: Python<'py>) -> PyResult<()> {
    let np = PyArrayModule::import(py)?;
    let dict = PyDict::new(py);
    dict.set_item("np", np.as_pymodule())?;
    let pyarray: &PyArray<i32> = py
        .eval("np.array([1, 2, 3], dtype='int32')", Some(&dict), None)?
        .extract()?;
    let slice = pyarray.as_slice().into_pyresult("Array Cast failed")?;
    assert_eq!(slice, &[1, 2, 3]);
    Ok(())
}

Write Python module by rust

Please see example directory for a complete example

[lib]
name = "rust_ext"
crate-type = ["cdylib"]

[dependencies]
numpy = "0.3"
ndarray = "0.11"

[dependencies.pyo3]
version = "^0.4.1"
features = ["extension-module"]

extern crate ndarray;
extern crate numpy;
extern crate pyo3;

use ndarray::{ArrayD, ArrayViewD, ArrayViewMutD};
use numpy::{IntoPyArray, IntoPyResult, PyArray, PyArrayModule};
use pyo3::prelude::{pymodinit, PyModule, PyResult, Python};

#[pymodinit]
fn rust_ext(py: Python, m: &PyModule) -> PyResult<()> {
    // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
    // You **must** write this sentence for PyArray type checker working correctly
    // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
    let _np = PyArrayModule::import(py)?;

    // immutable example
    fn axpy(a: f64, x: ArrayViewD<f64>, y: ArrayViewD<f64>) -> ArrayD<f64> {
        a * &x + &y
    }

    // mutable example (no return)
    fn mult(a: f64, mut x: ArrayViewMutD<f64>) {
        x *= a;
    }

    // wrapper of `axpy`
    #[pyfn(m, "axpy")]
    fn axpy_py(py: Python, a: f64, x: &PyArray<f64>, y: &PyArray<f64>) -> PyResult<PyArray<f64>> {
        let np = PyArrayModule::import(py)?;
        let x = x.as_array().into_pyresult("x must be f64 array")?;
        let y = y.as_array().into_pyresult("y must be f64 array")?;
        Ok(axpy(a, x, y).into_pyarray(py, &np))
    }

    // wrapper of `mult`
    #[pyfn(m, "mult")]
    fn mult_py(_py: Python, a: f64, x: &PyArray<f64>) -> PyResult<()> {
        let x = x.as_array_mut().into_pyresult("x must be f64 array")?;
        mult(a, x);
        Ok(())
    }

    Ok(())
}

Contribution

This project is in pre-alpha version. We need your feedback. Don't hesitate to open issues!

Version

• v0.3.0

  • Breaking Change: Migrated to pyo3 from rust-cpython
  • Some api addition
  • Static type checking with PhantomData

• v0.2.1

  • NEW: trait IntoPyErr, IntoPyResult for error translation

• v0.2.0

  • NEW: traits IntoPyArray, ToPyArray
  • MOD: Interface of PyArray creation functions are changed

• v0.1.1

  • Update documents

• v0.1.0

  • First Release
  • Expose unsafe interfase of Array and UFunc API