numpy.ufunc.reduceat

ufunc.reduceat(a, indices, axis=0, dtype=None, out=None)

Выполняет (локальное) сокращение со специфицированными срезами по одной оси.

Для i в range(len(indices)), reduceat вычисляет ufunc.reduce(a[indices[i]:indices[i+1]]), что становится i-й обобщенной «строкой», параллельной axis в конечном результате (например, в двумерной матрице, если axis = 0, она становится i-й строкой, но если axis = 1, она становится i-м столбцом). Существует три исключения из этого правила:

• когда i = len(indices) - 1 (так что для последнего индекса), indices[i+1] = a.shape[axis].
• если indices[i] >= indices[i + 1], i-я обобщенная «строка» просто a[indices[i]].
• если indices[i] >= len(a) или indices[i] < 0, возникает ошибка.

Форма выходных данных зависит от размера indices и может быть больше, чем a (это происходит, если len(indices) > a.shape[axis]).

Параметры:

a : array_like Массив для обработки. indices : array_like Парные индексы, разделенные запятыми (а не двоеточием), указывающие срезы для сокращения. axis : int, необязательно Ось, по которой применяется reduceat. dtype : код типа данных, необязательно Тип, используемый для представления промежуточных результатов. По умолчанию – тип данных выходного массива, если он предоставлен, или тип данных входного массива, если выходной массив не предоставлен. out : ndarray, None или кортеж ndarray и None, необязательно Место, в которое сохраняется результат. Если не предоставлен или None, создается новый массив. Для согласованности с ufunc.__call__, если задан в качестве ключевого аргумента, он может быть заключен в кортеж из одного элемента. Изменено в версии 1.13.0: Допускаются кортежи для ключевого аргумента.

Возвращает:

r : ndarray Значения, полученные после сокращения. Если out был предоставлен, r является ссылкой на out.

Примечания

Описание примера:

Если a является одномерным, функция ufunc.accumulate(a) эквивалентна ufunc.reduceat(a, indices)[::2], где indices представляет собой range(len(array) - 1) с нулём, помещенным в каждый другой элемент: indices = zeros(2 * len(a) - 1), indices[1::2] = range(1, len(a)).

Не обманывайтесь названием этого атрибута: reduceat(a) не обязательно меньше, чем a.

Примеры

Чтобы взять сумму последовательных четырех значений:

>>> np.add.reduceat(np.arange(8),[0,4, 1,5, 2,6, 3,7])[::2]
array([ 6, 10, 14, 18])

Пример для 2-мерного случая:

>>> x = np.linspace(0, 15, 16).reshape(4,4)
>>> x
array([[  0.,   1.,   2.,   3.],
       [  4.,   5.,   6.,   7.],
       [  8.,   9.,  10.,  11.],
       [ 12.,  13.,  14.,  15.]])

# reduce such that the result has the following five rows:
# [row1 + row2 + row3]
# [row4]
# [row2]
# [row3]
# [row1 + row2 + row3 + row4]

>>> np.add.reduceat(x, [0, 3, 1, 2, 0])
array([[ 12.,  15.,  18.,  21.],
       [ 12.,  13.,  14.,  15.],
       [  4.,   5.,   6.,   7.],
       [  8.,   9.,  10.,  11.],
       [ 24.,  28.,  32.,  36.]])

# reduce such that result has the following two columns:
# [col1 * col2 * col3, col4]

>>> np.multiply.reduceat(x, [0, 3], 1)
array([[    0.,     3.],
       [  120.,     7.],
       [  720.,    11.],
       [ 2184.,    15.]])