回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我一直在疯狂地想弄明白我在这里做错了什么蠢事。</p>
<p>我使用的是NumPy,我有一些特定的行索引和列索引要从中选择。以下是我问题的要点:</p>
<pre><code>import numpy as np
a = np.arange(20).reshape((5,4))
# array([[ 0, 1, 2, 3],
# [ 4, 5, 6, 7],
# [ 8, 9, 10, 11],
# [12, 13, 14, 15],
# [16, 17, 18, 19]])
# If I select certain rows, it works
print a[[0, 1, 3], :]
# array([[ 0, 1, 2, 3],
# [ 4, 5, 6, 7],
# [12, 13, 14, 15]])
# If I select certain rows and a single column, it works
print a[[0, 1, 3], 2]
# array([ 2, 6, 14])
# But if I select certain rows AND certain columns, it fails
print a[[0,1,3], [0,2]]
# Traceback (most recent call last):
# File "<stdin>", line 1, in <module>
# ValueError: shape mismatch: objects cannot be broadcast to a single shape
</code></pre>
<p>为什么会这样?当然,我应该能够选择第一,第二,第四行,第一和第三列?我期望的结果是:</p>
<pre><code>a[[0,1,3], [0,2]] => [[0, 2],
[4, 6],
[12, 14]]
</code></pre>