Question

给出一个n维的numpy数组。现在给出一个轴和相应的索引。该特定轴索引中的所有元素都应替换为给定值。三维数组的示例：

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

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

给定axis = 1，index = 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,:]中选择轴= 1的index = 0并在所有其他轴上将值设置为0

替换n维numpy数组中特定轴索引的元素

2 个答案: