Function partition 

pub fn partition(a: &Array, kth: i32, axis: impl Axis) -> Array

The array partitioned around position kth along one axis; see Array::partition.