Bu, bir kod satırındaki numpy dizisine ek bir boyut eklemenin en iyi yolu mu?

Question

Eğer k keyfi bir şeklin numpy dizisi ise, bu yüzden k.shape = (s1, s2, s3, ..., sn), ve k.shape(s1, s2, ..., sn, 1) olacak şekilde yeniden şekillendirmek istiyorum, bu bir satırda bunu yapmanın en iyi yoludur?

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

Answer 1

Böyle kolay:

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

Ama istediğin sonunda boş bir boyut eklemek için ise, kullanmalısınız

numpy.newaxis:

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

veya

k = k[..., None] 

(Bkz. documentation on slicing).

Answer 2

python3.5 itibaren, np.ndarray.shapenp.reshape içeride ile açma kullanabilirsiniz:

>>> my_array = np.array([1,2,3,4]) 
>>> my_array.reshape(1, *my_array.shape) 
array([[1, 2, 3, 4]]) 
>>> my_array.reshape(*my_array.shape, 1) 
array([[1], 
     [2], 
     [3], 
     [4]]) 

my_array.reshape(1, *my_array.shape) eser eski sürümlerinde de.