“如果” Python 中用于数组的函数

Question

我有这个函数：

def q(x,y):
    r2 = (x/2.)**2 + (2.0*y)**2
    if (r2 > 0.04): 
        return 0.
    else:
        return 1.5

我想使用类似的东西来调用它

from pylab import *
dl=0.025
X, Y = mgrid[-0.5:0.5:dl, -0.5:0.5:dl]
g(X,Y)

，但显然这会在比较中产生错误。

是否可以在无需为 X 和 Y 进行 for 循环的情况下完成此操作？ 因为如果我想为两个双精度调用 q(x,y)，则必须针对这种情况重新实现 for 循环...

编辑： （将其添加到问题中，因为它对于评论而不是答案，但它可能会帮助其他人回答。）

看来 pylab.mgrid 与 numpy.mgrid 相同。

针对numpy进行调整，此代码

import numpy
def q(x,y):
    r2 = (x/2.)**2 + (2.0*y)**2
    if (r2 > 0.04): 
        return 0.
    else:
        return 1.5

dl=0.025
X, Y = numpy.mgrid[-0.5:0.5:dl, -0.5:0.5:dl]
q(X,Y)

给出了此错误

    if (r2 > 0.04):
ValueError: The truth value of an array with more than one element is
ambiguous. Use a.any() or a.all()

Answer 1

表达式r2> 0.04 计算结果为布尔值的 NumPy 数组，因此您不能在 if 语句中使用它。不过，您可以将其自动重新解释为数字：

def q(x,y):
    r2 = (x/2.)**2 + (2.0*y)**2
    return 1.5 * (r2 <= 0.04)

Answer 2

您可以使用 map 将函数应用于每对元素。