擅长:python、mysql、java
<p>您尝试使用<code>1</code>作为第一个索引3次,<code>5</code>作为第二个维度的索引(同样是3次)。这将使元素在<code>A[1,5,:]</code>处重复<em>三次</em></p>
<pre><code>A = np.random.rand(6,6,6);
B = A[[1,1,1], [5,5,5]]
# [[ 0.17135991, 0.80554887, 0.38614418, 0.55439258, 0.66504806, 0.33300839],
# [ 0.17135991, 0.80554887, 0.38614418, 0.55439258, 0.66504806, 0.33300839],
# [ 0.17135991, 0.80554887, 0.38614418, 0.55439258, 0.66504806, 0.33300839]]
B.shape
# (3, 6)
</code></pre>
<p>相反,您需要为矩阵的每个<em>轴指定<code>[1,5]</code></p>
<pre><code>A[[1,5], [1,5], [1,5]] = [5, 10]
</code></pre>