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

Question

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

例如，在答案中，我使用了s=np.argsort(u)，然后对“虚拟”排序的u[s]进行了所有计算。我遇到过那种情况，那么我需要一个布尔值的数组，给出类似于u[s][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__。