python - 从不分配 numpy.ufunc 的输出

Question

这个问题有关于使用输入作为输出来使用 numpy.ufunc 计算某些东西的信息:

• Numpy passing input array as `out` argument to ufunc

是否可以避免为 numpy.ufunc 的不需要的输出分配空间？例如，假设我只想要来自 modf 的两个输出之一。我可以确保根本不分配另一个不需要的数组吗？

我认为将 _ 传递给 out 可能会成功，但它会引发错误:

import numpy as np
ar = np.arange(6)/3
np.modf(ar, out=(ar, _))    

TypeError: return arrays must be of ArrayType

正如文档中所说，传递 None 意味着输出数组在函数中分配并返回。我可以忽略返回值，但它仍然必须在函数内部分配和填充。

Answer 1

您可以通过传递“假”数组来最小化分配:

ar = np.arange(6) / 3
np.modf(ar, ar, np.broadcast_arrays(ar.dtype.type(0), ar)[0])

这个虚拟数组与单个double 一样大，modf 不会在内部进行分配。

编辑 根据@Eric 和@hpaulj 的建议，一个更通用和更长期的解决方案是

np.lib.stride_tricks._broadcast_to(np.empty(1, ar.dtype), ar.shape, False, False)