SyntaxError:无法为Python值的区间赋值给函数调用、Dirac函数

Question

我试图实现一个简单的函数，对于某个值返回1，这里是0。你知道吗

但我一直有个错误：

Diracx(i) = 1
SyntaxError: can't assign to function call

你能帮忙吗？你知道吗

def Dirac(x):
    Diracx = np.zeros(size(x))
    for i in range(size(x)):
        if (x(i) == 0):
            Diracx(i) = 1    

    return Diracx 


plt.plot(x,DiracDelta(x))

Answer 1

或者，如果只是为了刺激别人提供更好的答案。你知道吗

>>> x = np.array([-2,-1,0,1,2])
>>> a = np.zeros(np.size(x))
>>> a[x!=0]=1
>>> a
array([ 1.,  1.,  0.,  1.,  1.])