Заменить элементы определенного индекса оси в n-мерном массиве NumPy

Question

Учитывая n-мерный массив NumPy.Теперь дана ось и соответствующий индекс.Все элементы в этом конкретном индексе оси должны быть заменены данным значением.Пример трехмерного массива:

>>a = np.ones((2,2,2))
array([[[ 1.,  1.],
        [ 1.,  1.]],

        [[ 1.,  1.],
        [ 1.,  1.]]])

Заданная ось = 1, индекс = 0.Все элементы в этом индексе оси должны быть равны нулю.

>>a
array([[[ 0.,  0.],
        [ 1.,  1.]],

       [[ 0.,  0.],
        [ 1.,  1.]]])

Answer 1

Использование swapaxes:

a.swapaxes(0, axis)[index] = value

Пример:

>>> import numpy as np
>>> a = np.zeros((2,3,4))
>>> a.swapaxes(0, 1)[2] = 3
>>> a
array([[[0., 0., 0., 0.],
        [0., 0., 0., 0.],
        [3., 3., 3., 3.]],

       [[0., 0., 0., 0.],
        [0., 0., 0., 0.],
        [3., 3., 3., 3.]]])

Answer 2

вы можете сделать a[:,0,:] = 0, и вы получите свой вывод, где в a[:,0,:] вы выбираете index = 0 оси = 1 и устанавливаете значение 0 по всей другой оси