目前,大多数numpy函数会自动删除单例维度,为了保留它们,需要记住各种参数,例如keepdims=True
或np.newaxis
。我想知道是否已经(或将要)全球设置,因为我正在编写一个需要始终保持维度的程序。我理想的代码就像
import numpy as np
np.setparams(keepdims=True) # anything like this?
x = np.random.randint(10, size=(5, 10))
np.sum(x, axis=0).shape # prints (1, 10)
x[:, 0].shape # prints (5, 1)
谢谢!