我的代码有问题,该代码是一种用于数值求解谐波振荡器的 Runge Kutta 算法。不幸的是,我收到一个错误,说我不能将序列乘以浮点类型的非整数。考虑到这段代码几乎是来自教育文本的逐字记录,我看不出错误在哪里。有人可以帮我吗?感谢您友好的邻居蜘蛛侠。
import numpy as np
import matplotlib.pyplot as plt
#Variable Definitions
N = 500
x0 = 1.0
v0 = 0.0
dur = 10.0
dt = dur/float(N-1)
#creating the array
y = np.zeros([N,2])
y[0,0] = x0
y[0,1] = v0
#runge kutta algorithm
def rk4(y, time, dt, deriv)
k1 = dt * deriv(y,time)
k2 = dt * deriv(y + 0.5*k1,time + 0.5*dt)
k3 = dt * deriv(y + 0.5*k2, time + 0.5*dt)
k4 = dt * deriv(y + k3, time + dt)
y_next = y + (k1 + 2*(k2+k3)+k4)/6
return y_next
#Harmonic oscillator
def Harmonic(x,time):
y0 = x[1]
y1 = (-1)*x[0]
return ([y0,y1])
#forming data points
for i in range(N-1):
y[i+1] = rk4(y[i],0, dt, Harmonic)
time = np.linspace(0, dur, N)
#plotting
plt.plot(time,y[:,1])
plt.show()
错误在第 33 行。
TypeError:不能将序列乘以“float”类型的非整数