前提・実現したいこと
重積分のグラフを描画したいのですがエラーが出て描画できません。
発生している問題・エラーメッセージ
TypeError: can't convert complex to float
該当のソースコード
python
1import numpy as np 2from scipy import integrate 3import sympy as sym 4import math 5import matplotlib.pyplot as plt 6import cmath 7 8E0 = 1 9f = 3*10**9 10c = 3*10**8 11lum = c / f 12k = 2*math.pi / lum 13 14Nf = np.arange(0,20,0.5) 15 16ds = 100 17dp = 100 18 19W = [] 20L = [] 21 22x = -1 23y = 1 24 25xs = 0 26ys = 0 27 28xp = 0 29yp = 0 30 31n = 1 32 33rs = np.sqrt(ds**2 - 2*xs*x + x**2 - 2*ys*y + y**2) 34rp = np.sqrt(dp**2 - 2*xp*x + x**2 - 2*yp*y + x**2) 35 36Q = (n*rp / (abs(n)*abs(rp))) + (n*rs / (abs(n)*abs(rs))) / 2 37 38intE = [] 39EP1 = Q*((np.exp(-1j*k*(rs+rp)))/(rs*rp)) 40for i in range(len(Nf)): 41 w = np.sqrt((4*lum*ds*dp*Nf[i])/(ds+dp)) 42 l = w 43 W.append(w) 44 L.append(l) 45a = sym.symbols('a') 46b = sym.symbols('b') 47for j in range(len(Nf)): 48 inte = sym.integrate(EP1,(a,-W[j]/2,W[j]/2),(b,-L[j]/2,L[j]/2)) 49 intE.append(inte) 50 51print(intE) 52 53EP = [] 54for k in range(len(Nf)): 55 EP2 = (1j*E0 / lum) * intE[k] 56 EP.append(EP2) 57print(EP) 58print(EP[2]) 59 60fig = plt.figure() 61ax = fig.add_subplot(111) 62ax.plot(Nf,EP) 63ax.set_xlabel('Nf') 64ax.set_ylabel('E(P)') 65plt.legend(loc='best') 66plt.grid()
試したこと
複素数を扱っているためcmathモジュールを使ったり、配列を使って数値を格納したりしたのですが、特に変化は起こりませんでした。
EPをprintしたら中身はありました。
補足情報(FW/ツールのバージョンなど)
Python3.7.6
Spyder
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2020/09/16 04:04