我现在想做的是:
x = x[:, None, None, None, None, None, None, None, None, None]
基本上,我想将我的 Numpy 数组扩展 9 个维度。或者 N 维数,其中 N 可能无法事先知道!
有更好的方法吗?
最佳答案
另一种方法是使用 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)
关于python - 是否有更 Pythonic/优雅的方式来扩展 Numpy 数组的维度?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40069220/