我试图在 Python 2.7 代码中绘制 Dirac Delta 矩形函数,以便:
enter image description herefrom __future__ import division
import numpy as np
import matplotlib.pyplot as plt
def ddf(x,sig):
if -(1/(2*sig))<=x and x<=(1/(2*sig)):
val=sig
else:
val=0
return val
X=np.linspace(-5,5,1000)
for sig in np.arange(1,5,0.1):
plt.cla()
plt.grid()
plt.title('Dirac Delta function',size=20)
plt.xlabel('X values',size=10)
plt.ylabel("Dirac Delta functions' values",size=10)
plt.ylim(0,1)
plt.plot(X,ddf(X,sig),color='black')
plt.pause(0.5)
plt.show()
但是当我运行代码时,它给出了错误:
Traceback (most recent call last):
File "c:/Users/Shubhadeep/Desktop/dff.py", line 22, in <module>
plt.plot(X,ddf(X,sig),color='black')
File "c:/Users/Shubhadeep/Desktop/dff.py", line 7, in ddf
if -(1/(2*sig))<=x and x<=(1/(2*sig)):
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
任何人都可以解决这个问题?
问题是在函数中,您正在将 listx
与 floatsig
进行比较。一种解决方案是稍微修改函数,以便它逐个评估x
中的值,然后将评估附加到将由函数返回的新列表中:
def ddf(x,sig):
val = []
for i in x:
if -(1/(2*sig))<=i and i<=(1/(2*sig)):
val.append(sig)
else:
val.append(0)
return val
本站系公益性非盈利分享网址,本文来自用户投稿,不代表边看边学立场,如若转载,请注明出处
评论列表(54条)