Python中用于数组的“If”函数

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 将您的函数应用于每一对元素。