Shop mehrdimensionale numpy Array-Slice mit newaxis zum Objekt

Question

ich einen Code haben, wo ich immer wieder wiederholt ausgestrahlt Arrays auf komplexe Weise zu, zum Beispiel:

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

Gibt es ein Objekt, an dem ich diese Aufschneiden Spezifikationen speichern ?

dh (aber offensichtlich das nicht funktioniert):

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

---

Edit: vielleicht könnte dies mit numpy.broadcast_to getan werden, aber es ist unklar, wie genau gleichzeitig aber dafür sorgen, dass die richtigen Achsen ausgestrahlt ...

Answer 1

Sie können den Index Tupel manuell konstruieren, aber NumPy umfasst ein helper dafür:

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

Dann entspricht b[np.newaxis, ..., :, np.newaxis]b[slicetuple].

---

Als Referenz würde das Tupel Konstruktion manuell (np.newaxis, Ellipsis, slice(None), np.newaxis) sein. Auch np.newaxis is None, also (None, Ellipsis, slice(None), None) wäre gleichwertig.

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

s_ = IndexHelper()