import numpy as np
def romberg(f,l):
val=np.zeros(l)
for i in range(0,l):
h=2**(1-i)
N=2/h
val[i]=trapez(f,h,N)
return val
def trapez(f,h,N):
result=0
stuetz=np.zeros(N+1);
for j in range(0,N+1):
stuetz[j]=0+j/N
sum=0
for k in range(1,N): #bilde summe von 1-N-1
sum+=np.polyval(f,stuetz[k])
a=np.polyval(f,0)*0.5*h
b=(np.polyval(f,stuetz[N]))*0.5*h
result=a+(h*sum)+b
return result
主:
p=[1,0,0]
romberg(p,5)
错误:
----> 3 romberg(p,5)
----> 7 val[i]=trapez(f,h,N)
TypeError:' float' object不能解释为整数
我该如何解决这个请帮帮我? 我很感激!
答案 0 :(得分:5)
将N=2/h
更改为N=2//h
以执行整数除法。
def romberg(f,l):
val=np.zeros(l)
for i in range(0,l):
h=2**(1-i)
N=2//h
val[i]=trapez(f,h,N)
return val
修改强>
您也可以使用int(2/h)
def romberg(f,l):
val=np.zeros(l)
for i in range(0,l):
h=2**(1-i)
N=int(2/h)
val[i]=trapez(f,h,N)
return val