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

Posted 2023-02-14

【中文标题】这是在一行代码中向 numpy 数组添加额外维度的最佳方法吗？

【英文标题】：Is this the best way to add an extra dimension to a numpy array in one line of code?

【发布时间】：2013-07-23 23:52:05

【问题描述】：

如果 k 是一个任意形状的 numpy 数组，那么k.shape = (s1, s2, s3, ..., sn)，我想重塑它，使k.shape 变成(s1, s2, ..., sn, 1)，这是最好的方法吗？

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

【参考方案1】：

这样更容易：

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

但如果你只想在最后添加一个空维度，你应该使用numpy.newaxis：

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

或

k = k[..., None]

（见documentation on slicing）。

【参考方案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]])

【讨论】：

注意：np.reshape 比 np.expand_dims 快。