Поиск диапазона элементов массива numpy

У меня есть массив NumPy размером 94 х 155:

a = [1  2  20  68  210  290..
     2  33 34  55  230  340..
     .. .. ... ... .... .....]

Я хочу рассчитать диапазон каждой строки, чтобы в результате я получил 94 диапазона. Я попытался найти функцию numpy.range, которой, я думаю, не существует. Если это можно сделать через цикл, это тоже хорошо.

Я ищу что-то вроде numpy.mean, который, если мы установим параметр оси в 1, возвращает среднее значение для каждой строки в N-мерном массиве.

Ответ 2

Попробуйте следующее:

def range(x, axis=0):
    return np.max(x, axis=axis) - np.min(x, axis=axis)