擅长:python、mysql、java
<p>可以为前两个维度构建相应的索引数组。基本上是:</p>
<pre><code>[0 1 2]
[0 1 2]
[0 1 2]
[0 0 0]
[1 1 1]
[2 2 2]
</code></pre>
<p>您可以使用meshgrid函数构造这些。我将它们存储为m1,在示例中存储为m2:</p>
<pre><code>vals = np.arange(3*3*5).reshape(3, 3, 5) # test sample
m1, m2 = np.meshgrid(range(3), range(3), indexing='ij')
m3 = np.array([[0, 0, 1], 0, 0, 1], [1, 4, 4]])
sel_vals = vals[m1, m2, m3]
</code></pre>
<p>结果的形状与索引数组m1、m2和m3的形状匹配。你知道吗</p>