从函数创建Numpy

我需要这样的东西

def qqq(i,j):
    if i+j>2:
        return 0.5
    else:
        return 0
n=3
dcdt=np.fromfunction(lambda i,j: qqq(i,j)*i*j, (n,n), dtype=int)

---

但qqq更复杂。但它会导致错误“具有多个元素的数组的真值不明确。请使用a.any（）或a.all（）。我知道问题是函数只调用一次。如何使用函数中的“if elif else”结构创建这样的数组？

您应该将您的

qqq

函数转换为如下内容：

def qqq(i, j):
    return np.where(i + j > 2, 0.5, 0)

有关详细信息，请参阅