擅长:python、mysql、java
<p>下面的代码遵循传统的python风格,但没有针对速度进行优化。试试看它是否适合您的用例。</p>
<pre><code>A = [[1, 0, 0, 6, 1],
[8, 2, 1, 1, 5],
[1, 1, 0, 0, 4],
[3, 8, 7, 1, 2],
[3, 9, 1, 5, 5]]
b = [0, 1, 2, 3, 4]
result = [A[i][b_i] for i, b_i in enumerate(b)]
# -> [1, 2, 0, 1, 5]
</code></pre>