如何引用 numpy 数组的切片/视图

Question

我给earlier 的回答向我提出了一个问题：是否可以在不重复一堆括号的情况下引用 numpy 数组的视图或切片？

例如，在答案中，我使用了s=np.argsort(u)，然后对“虚拟”排序的u[s] 进行了所有计算。我曾经遇到过需要该数组的布尔掩码的情况，给出类似于u[s][mask] 的东西。对于更大的数据，我可能有一个 mask of a mask of a mask 。 . .直到事情开始看起来像是史酷比剧集的结尾。

但是，如果我将该数组分配给变量b=a[s][mask] 并更改b，a 不会更改，因此我最终在计算中携带了一堆括号。我已经尝试过uv=u.view()[s] 的各种安排，但似乎.view() 只能查看整个数组。我还缺少另一种方法吗？

Answer 1

您可能无法解决u[s] 的简单情况，但在u[s][mask] 等更复杂的情况下，您可以：

t = s[mask]
u[t] # same as u[s][mask]

也就是说，您可以将掩码简化为单个变量，但您可能无法完全摆脱它，除非您可能想用__getitem__ 和__setitem__ 编写自己的包装类。