擅长:python、mysql、java
<p>请尝试运行以下代码</p>
<pre><code>import numpy as np
x = [[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8],
[ 9, 10, 11]]
index_array = [0, 2, 1, 2]
index = 0
result = []
for item in x:
result.append(item[index_array[index]])
index += 1
print (result)
</code></pre>
<p>结果如下</p>
<pre><code>[0, 5, 7, 11]
>
</code></pre>