Numpy在全球范围内保持维度

时间:2018-05-09 08:28:51

标签: python numpy

目前,大多数numpy函数会自动删除单例维度,为了保留它们,需要记住各种参数,例如keepdims=Truenp.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)

谢谢!

0 个答案:

没有答案