这是在一行代码中为numpy数组添加额外维度的最佳方法吗？

Question

如果k是任意形状的numpy数组,那么k.shape = (s1, s2, s3, ..., sn),我想重塑它以便k.shape变成(s1, s2, ..., sn, 1),这是在一行中执行它的最佳方法吗？

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

Answer 1

它更容易这样:

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

但如果你想要的只是在最后添加一个空维度,你应该使用numpy.newaxis:

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

要么

k = k[..., None]

(参见切片文档).

Answer 2

您可以使用numpy.expand_dims

In [4]: import numpy as np                                                          

In [5]: a = [1,2,3,4]                                                                              

In [6]: np.expand_dims(a, axis=1)                                                                  
Out[6]: 
array([[1],
       [2],
       [3],
       [4]])