логическая ошибка в линейной интерполяции в питоне - PullRequest
1 голос
/ 07 мая 2019

Моя линейная интерполяция имеет некоторую логическую ошибку, она работает в определенных случаях, но не полностью работает.

Я пытался использовать другой способ написания логики для случаев экстраполяции.

def interpolate(x, y, x_test):
    for i in range(len(x)):
        if x[i] > x_test:   #extrapolated condition: when the largest value of
            x_below = i - 1 #list x is greater than x_test 
            x_above = i 
            y_below = i - 1
            y_above = i
            break
        elif x[i] < x_test: #extrapolated condition: when the largest value of 
            x_below = i + 1 #list x is greater than x_test 
            x_above = i 
            y_below = i + 1
            y_above = i
            break                
        else:             #interpolated condition: when x_test lies between  
            return y[i]    #two sample points.

    #a = (yabove - ybelow) / (xabove - xbelow)         
    a = (y[y_above] - y[y_below]) / (x[x_above] - x[x_below])  
    #b = ybelow - a * xbelow
    b = y[y_below] - a * x[x_below]
    #y’ = a * x’ + b
    return a * x_test + b  

interpolate([1, 3, 5], [1, 9, 25], 5.0) Я ожидаю, что выходной сигнал равен 25, но фактический выходной сигнал равен 17,0.

1 Ответ

0 голосов
/ 07 мая 2019

Я думаю, вы ищете что-то вроде этого:

def interpolate(x, y, x_test):
    for i in range(len(x)):
        if x[i] > x_test:   #extrapolated condition: when the largest value of
            x_below = i - 1 #list x is greater than x_test
            x_above = i
            y_below = i - 1
            y_above = i
            continue # <---- I changed break to continue
        elif x[i] < x_test: #extrapolated condition: when the largest value of
            x_below = i + 1 #list x is greater than x_test
            x_above = i
            y_below = i + 1
            y_above = i
            continue # <---- I changed break to continue
        else:             #interpolated condition: when x_test lies between
            return y[i]    #two sample points.

    #a = (yabove - ybelow) / (xabove - xbelow)
    a = (y[y_above] - y[y_below]) / (x[x_above] - x[x_below])
    #b = ybelow - a * xbelow
    b = y[y_below] - a * x[x_below]
    #y’ = a * x’ + b
    return (a * x_test + b)

print(interpolate([1, 3, 5], [1, 9, 25], 5.0))

Выход:

25

Обратите внимание, я изменил breaks на continues.

...