python - 如何获得恒定函数以在 NumPy 中保持形状

Question

我有一个形状NumPy数组，想通过某个函数运行所有元素。对于非常量函数，例如广播或广播工作得非常好，并返回预期的结果。对于，将函数应用于我的数组但是只返回标量1。A(m,n)ff(x) = xf(x) = x**2f(x) = 1A

有没有办法强制广播保持形状，即在这种情况下返回一个 1 的数组？

Answer 1

F(x) = 1不是一个函数，您需要使用defor创建一个函数lambda并返回 1。然后使用np.vectorize该函数在您的数组上应用该函数。

>>> import numpy as np
>>> f = lambda x: 1
>>> 
>>> f = np.vectorize(f)
>>> 
>>> f(np.arange(10).reshape(2, 5))
array([[1, 1, 1, 1, 1],
       [1, 1, 1, 1, 1]])

Answer 2

这听起来像是一份工作np.ones_like，或者np.full_like在一般情况下：

def f(x):
    result = np.full_like(x, 1)  # or np.full_like(x, 1, dtype=int) if you don't want to
                                 # inherit the dtype of x
    if result.shape == 0:
        # Return a scalar instead of a 0D array.
        return result[()]
    else:
        return result

Answer 3

使用x.fill(1). 确保正确返回它，因为fill它不会返回新变量，它会修改x