在数组的特定维度中设置所有值的Numpy方法

Question

假设我在matlab中有这个矩阵：

a(:,:,1) =

     1     2     3
     4     5     6
     7     8     9


a(:,:,2) =

    10    20    30
    40    50    60
    70    80    90

现在我可以了

^{pr2}$

我得到了

a(:,:,1) =

     1     2     3
     4     5     6
     7     8     9


a(:,:,2) =

     0     0     0
     0     0     0
     0     0     0

在python中使用numpy复制这种行为（a（：，：，2）=某个数组）的唯一方法是使用好的旧循环

for dim0 in range(a.shape[0]):
    for dim1 in range(a.shape[1]):
        a[dim0, dim1, 1] = 0

有没有更好的裸体方式来做这件事？在

Answer 1

是的。和Matlab中的方法完全一样。循环的作用与：

a[:,:,1] = 0

你可以阅读更多关于它的in the docs。在