rank_array

tsml_eval.utils.functions.rank_array(arr, higher_better=True)[source]

Assign a rank to each value in a 1D numpy array.

A lower rank number is assumed to be better. Lower values can receive better ranks or vice versa based on the higher_better parameter. Equal values receive the average of the ranks they would cover.

Parameters:
arrnumpy.ndarray

The input 1D array containing values to be ranked.

higher_betterbool, default=True

If True, lower values receive better ranks. If False (default), higher values receive better ranks.

Returns:
ranksnumpy.ndarray

Array of ranks, same shape as arr.