jaxvacua.cytools_interface.remove_zeros

Contents

jaxvacua.cytools_interface.remove_zeros#

remove_zeros(a, axis=1)#

Removes zero rows from a numpy array.

Parameters:

  • a (np.ndarray) – Input array.

  • axis (int, optional) – Axis along which to check for zero rows. Default is 1.

Returns:

tuple – A tuple containing the array with zero rows removed and a boolean flag array indicating which rows were kept.

Contents