語法錯誤：無法分配給函數調用，狄拉克函數值的INTERVALL的Python

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.])