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

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

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

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

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

Ответы

Ответ 1

Я думаю, что np.ptp может делать то, что вы хотите:

http://docs.scipy.org/doc/numpy/reference/generated/numpy.ptp.html

r = np.ptp(a,axis=1)

где r - ваш массив массивов.

Ответ 2

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

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