我试图用scipy库,特别是rbf('cubic')来插值缺失值。但我得到了以下错误。
Traceback (most recent call last):
File "C:\Users\St\Desktop\py_magn\inter.py", line 89, in <module>
rbfp = Rbf(xn ,yn, magn, function='cubic') #PARAMETERS
File "C:\Users\St\AppData\Local\Programs\Python\Python38\lib\site-packages\scipy\interpolate\rbf.py", line 239, in __init__
self.epsilon = np.power(np.prod(edges)/self.N, 1.0/edges.size)
ZeroDivisionError: float division by zero
my code is the following:
x, y, mag = df[:,0], df[:,1], df[:,3]
emptyInd = np.where(np.isnan(mag))
print(emptyInd[0])
#----------------------------------------------------------------------------------------------------
def eucl_dist(pnt,x,y,mag):
value = []
dist = 0.0
typ = [('Eucli Dist', float), ('x', float), ('y', float), ('Magn', float)]
for i in range(len(x)):
dist = sqrt((pnt[0]-x[i])**2 + (pnt[1]-y[i])**2)
res = [dist, x[i], y[i], mag[i]]
value.append(res)
value = np.vstack(value)
return value,typ
#--------------------------------------------------------------------------------
xinter=[]
yinter=[]
magInter=[]
neigh = []
if len(emptyInd)!=0 :
listOfEMPval = list(zip(emptyInd[0])) #, emptyInd[1]))
for ind in listOfEMPval:
xn=[]
yn=[]
magn=[]
xinter = np.take(x,ind)
yinter = np.take(y,ind)
edist, typ = eucl_dist((xinter, yinter), x, y, mag)
edist = rf.unstructured_to_structured(edist, np.dtype(typ))
indx = np.argsort(edist, order='Eucli Dist')
edist = np.reshape([edist[i] for i in indx],(len(edist), 1))
sz=182
for k in range(sz):
if np.isnan(edist[k][0][3]):
sz+=1
else:
xn = np.append(xn, edist[k][0][1])
yn = np.append(yn, edist[k][0][2])
magn = np.append(magn, edist[k][0][3])
rbfp = Rbf(xn ,yn, magn, function='cubic') #PARAMETERS
magInter = np.append(magInter, rbfp(xinter,yinter)) #INTERPOLATION
for i in range(len(listOfEMPval)):
np.put(mag, listOfEMPval[i], magInter[i])
正如我在 "答案 "部分所写,我解决了我的问题。唯一的想法是,是否有任何方法使它更快。