python - 是否有更 Pythonic/优雅的方式来扩展 Numpy 数组的维度？

Question

我现在想做的是:

x = x[:, None,  None,  None,  None,  None,  None,  None,  None,  None]

基本上，我想将我的 Numpy 数组扩展 9 个维度。或者 N 维数，其中 N 可能无法事先知道!

有更好的方法吗？

Answer 1

另一种方法是使用 reshaping -

x.reshape((-1,) + (1,)*N)  # N is no. of dims to be appended

因此，基本上对于与单一维度对应的None's，我们沿这些维度使用长度为1 的形状。对于第一个轴，我们使用 -1 的形状来将所有元素插入其中。

sample 运行-

In [119]: x = np.array([2,5,6,4])

In [120]: x.reshape((-1,) + (1,)*9).shape
Out[120]: (4, 1, 1, 1, 1, 1, 1, 1, 1, 1)