Ist dies der beste Weg, um fügen Sie eine zusätzliche dimension, um ein numpy-array in nur einer Zeile code?

Wenn k ein numpy-array mit einer beliebigen Form, also k.shape = (s1, s2, s3, ..., sn), und das möchte ich Umformen, so dass k.shape wird (s1, s2, ..., sn, 1), ist dies der beste Weg, es zu tun in einer Zeile?

k.reshape(*(list(k.shape) + [1])

InformationsquelleAutor immarried | 2013-07-24

  1. 30

    Es ist einfacher wie dies:

    k.reshape(k.shape + (1,))

    Aber wenn alle Sie wollen, ist das hinzufügen einer leeren dimension am Ende, die Sie verwenden sollten numpy.newaxis:

    import numpy as np
k = k[..., np.newaxis]

    oder 

    k = k[..., None]

    (Siehe Dokumentation auf das schneiden).

    InformationsquelleAutor tiago

Schreibe einen Kommentar