python - Numpy 天花板和地板 "out"参数

Question

来自 NumPy docs for ceil , numpy.ceil 函数有两个参数，第二个是 out。文档没有说明此 out 参数的作用，但我假设您可以设置此函数返回的输出类型，但我无法让它工作:

In [107]: np.ceil(5.5, 'int')
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-107-c05bcf9f1522> in <module>()
----> 1 np.ceil(5.5, 'int')

TypeError: return arrays must be of ArrayType

In [108]: np.ceil(5.5, 'int64')
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-108-0937d09b0433> in <module>()
----> 1 np.ceil(5.5, 'int64')

TypeError: return arrays must be of ArrayType

是否可以使用此参数使 np.ceil 返回一个整数？

谢谢。

Answer 1

out 是输出数组(必须与输入具有相同的形状)。

如果您将其构造为所需的dtype，那将是您得到的dtype:

>>> arr = np.array([5.5, -7.2])
>>> out = np.empty_like(arr, dtype=np.int64)
>>> np.ceil(arr, out)
array([ 6, -7], dtype=int64)
>>> out
array([ 6, -7], dtype=int64)