Запись вдоль оси в NumPy

Функция put_along_axis по форме похожа на взятие вдоль оси, но подставляет новое значение в указанные позиции и меняет исходный массив на месте. Массив индексов должен иметь ту же число размерностей, что и таблица.

В двух строках заменим крайние ячейки на 99 и выведем таблицу:

import numpy as np

table = np.array([[1, 2, 3], [4, 5, 6]])
cols = np.array([[0], [2]])
np.put_along_axis(table, cols, 99, axis=1)
print(table) # выведет [[99  2  3] [ 4  5 99]]

В таблице [[10, 20], [30, 40]] запишите 0 в первый столбец первой строки и второй столбец второй строки, затем выведите таблицу.

В таблице [[1, 2, 3], [4, 5, 6]] поставьте 7 в середину каждой строки по индексу 1 вдоль оси столбцов и выведите результат.