NumPy 中高级索引的结果形状

Question

当索引具有多个相同大小数组的 numpy 数组时，行为取决于索引轴的位置。

• 如果轴彼此相邻，则将轴添加到索引数组的位置。
• 如果轴不相邻，则将新轴添加到数组的开头。

import numpy as np
a = np.ones((4, 5, 6, 7, 8))
print(a[:, [0, 1], [0, 1], :, :].shape)       # (4, 2, 7, 8)
print(a[:, [0, 1], :, [0, 1], :].shape)       # (2, 4, 6, 8)
print(a[:, [0, 1], [0, 1], [0, 1], :].shape)  # (4, 2, 8)
print(a[:, [0, 1], [0, 1], :, [0, 1]].shape)  # (2, 4, 7)

我的问题与this question 和答案中提到的PR 有关。 不过好像这个oindex还没有。

我有哪些选择才能实现一致的行为？例如，无论中间是否有:，始终在开头添加新轴。

Answer 1

您的示例清楚地说明了结合高级和基本索引产生的歧义。链接是一个不太明确的情况，涉及列表和标量。

有

a[:, [0, 1], :, [0, 1], :]

5 和 7 收缩为 2，但它应该使用什么顺序呢？

(4, 2, 6, 8)
(4, 6, 2, 8)

面对这种模棱两可的情况，numpy 开发人员选择将其放在首位

(2, 4, 6, 8)

从 NEP/PR 可以看出，这个问题并不简单，还没有解决。