numba

Module for numba equivalents of some unsupported numpy functions.

Functions

numba_all_axis1(x)

Numba compatible implementation of np.all(..., axis=1) for 2d arrays.

numba_all_axis2(x)

Numba compatible implementation of np.all(..., axis=2) for 3d arrays.

numba_any_axis1(x)

Numba compatible implementation of np.any(..., axis=1) for 2d arrays.

numba_choice_bool_axis0(valids)

For each row in valids, chooses a random index of the true elements in that row.

numba_delete_axis0(x, pos)

Numba compatible implementation of np.delete(..., axis=0) for 2d arrays.

numba_diff(x)

Numba compatible implementation of np.diff(...) for 1d arrays.

numba_diff_axis0(x)

Numba compatible implementation of np.diff(..., axis=0) for 2d arrays.

numba_insert(x, pos, value)

Numba compatible implementation of np.insert(...) for 2d arrays.

numba_insert_axis0(x, pos, value)

Numba compatible implementation of np.insert(..., axis=0) for 2d arrays.

numba_int2bool(arr, size)

Converts an ndarray of integers to a boolean representation.

numba_take_advanced(arr, idx[, out])

Numba compatible implementation of the advanced indexing scheme in numpy.