use ndarray::{Array, Axis, stack, arr2};
let a = Array::linspace(0., 1., 5).into_shape((5,1)).unwrap();
let b = Array::<f64, _>::ones((5, 1));
let res = stack(Axis(1), &[b.view(), a.view()]);
assert!(res == Ok(arr2(&[[1., 0.],
[1., 0.25],
[1., 0.5],
[1., 0.75],
[1., 1.]])));
It could be worth to have an example in ndarray_for_numpy_users section of the docs.
I was looking for an equivalent function of numpy's
column_stack
(see https://docs.scipy.org/doc/numpy/reference/generated/numpy.column_stack.html) inndarray
docs.This is how I managed to solve the issue:
It could be worth to have an example in
ndarray_for_numpy_users
section of the docs.