numpy.ufunc.at

ufunc.at(a, indices, b=None)

Выполняет неоперенную операцию на месте с операндом ‘a’ для элементов, указанных ‘indices’. Для операции сложения ufunc, этот метод эквивалентен a[indices] += b, за исключением того, что результаты накапливаются для элементов, которые индексируются более одного раза. Например, a[[0,0]] += 1 будет инкрементировать первый элемент один раз из-за буферизации, тогда как add.at(a, [0,0], 1) будет инкрементировать первый элемент два раза.

Введено в версии 1.8.0.

Параметры:

a : array_like Массив, над которым нужно выполнить операцию на месте. indices : array_like or tuple Массивоподобный объект индекса или объект среза для индексирования первого операнда. Если первый операнд имеет несколько измерений, индексы могут быть кортежем массивоподобных объектов индексов или объектов срезов. b : array_like Второй операнд для ufunc, требующий двух операндов. Операнд должен быть транслируемым над первым операндом после индексирования или среза.

Примеры

Установить элементы 0 и 1 в их отрицательные значения:

>>> a = np.array([1, 2, 3, 4])
>>> np.negative.at(a, [0, 1])
>>> print(a)
array([-1, -2, 3, 4])

Инкрементировать элементы 0 и 1, и инкрементировать элемент 2 дважды:

>>> a = np.array([1, 2, 3, 4])
>>> np.add.at(a, [0, 1, 2, 2], 1)
>>> print(a)
array([2, 3, 5, 4])

Добавить элементы 0 и 1 в первый массив ко второму массиву и сохранить результаты в первый массив:

>>> a = np.array([1, 2, 3, 4])
>>> b = np.array([1, 2])
>>> np.add.at(a, [0, 1], b)
>>> print(a)
array([2, 4, 3, 4])