存儲與newaxis多維numpy的數組切片對象

Question

我有一些代碼，我多次需要重複播放以複雜的方式排列，例如：

a = b[np.newaxis, ..., :, np.newaxis] * c[..., np.newaxis, np.newaxis, :] 

是否有對象，我可以存儲這些切片規格？

即（但顯然這不起作用）：

s1 = magic([np.newaxis, ..., :, np.newaxis]) 
s2 = magic([..., np.newaxis, np.newaxis, :]) 

---

編輯：也許這可能是與numpy.broadcast_to完成，但目前還不清楚，同時使究竟是如何確保正確的軸播完了...

Answer 1

您可以手動構建索引記錄，但與NumPy包括helper吧：

slice_tuple = np.s_[np.newaxis, ..., :, np.newaxis] 

然後b[np.newaxis, ..., :, np.newaxis]相當於b[slicetuple]。

---

作爲參考，手動構建元組將是(np.newaxis, Ellipsis, slice(None), np.newaxis)。另外，np.newaxis is None，所以(None, Ellipsis, slice(None), None)將是等同的。

class IndexHelper(object): 
    def __getitem__(self, arg): 
     return arg 

s_ = IndexHelper()