Numpy全局保持维度

hfyxw5xn 于 2023-10-19 发布在其他

关注(0)|答案(1)|浏览(93)

目前，大多数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)

谢谢你，谢谢！

numpy

来源：https://stackoverflow.com/questions/50248804/numpy-keep-dimensions-globally

1条答案

按热度按时间

5rgfhyps1#

根据你正在处理的numpy版本，你可以做的是研究感兴趣的函数的签名，目前是numpy.sum，找到numpy.sum._implementation.__defaults__中的默认值，并根据你的意愿改变它。

>>> import numpy as np
>>> a = np.arange(5)
>>> np.sum(a)
10
>>> (defaults := numpy.sum._implementation.__defaults__)
(None, None, None, False, <no value>, <no value>)
#                  ^^^^^
>>> numpy.sum._implementation.__defaults__ = (defaults[:3], True, defaults[4:])
>>> np.sum(a)
array([10])

但是，哪种解决方案不会影响numpy.ndarray.sum。

>>> a.sum()
10

赞(0）回复(0）举报 2023-10-19

我来回答

Numpy全局保持维度

1条答案

相关问题

热门标签

最新问答