Линейная интерполяция с использованием numpy.interp

У меня есть 1-мерный массив A поплавков, который в основном хорош, но некоторые из значений отсутствуют. Отсутствующие данные заменяются нан (а не числом). Я должен заменить недостающие значения в массиве линейной интерполяцией из близких хороших значений. Так, например:

F7(np.array([10.,20.,nan,40.,50.,nan,30.])) 

должен вернуться

np.array([10.,20.,30.,40.,50.,40.,30.]). 

Каков наилучший способ сделать это с помощью Python?

Любая помощь приветствуется

благодаря

python,numpy,interpolation,linear-interpolation,

10

Ответов: 3


13

Вы можете использовать scipy.interpolate.interp1d:

>>> from scipy.interpolate import interp1d
>>> import numpy as np
>>> x = np.array([10., 20., np.nan, 40., 50., np.nan, 30.])
>>> not_nan = np.logical_not(np.isnan(x))
>>> indices = np.arange(len(x))
>>> interp = interp1d(indices[not_nan], x[not_nan])
>>> interp(indices)
array([ 10.,  20.,  30.,  40.,  50.,  40.,  30.])

EDIT : мне потребовалось некоторое время, чтобы понять, как np.interpработает, но это тоже может сделать работу:

>>> np.interp(indices, indices[not_nan], x[not_nan])
array([ 10.,  20.,  30.,  40.,  50.,  40.,  30.])

Я бы пошел pandas. Минималистический подход с oneliner:

from pandas import *
a=np.array([10.,20.,nan,40.,50.,nan,30.])
Series(a).interpolate()   

Out[219]:
0    10
1    20
2    30
3    40
4    50
5    40
6    30

Или если вы хотите сохранить его как массив:

Series(a).interpolate().values

Out[221]:
array([ 10.,  20.,  30.,  40.,  50.,  40.,  30.])

0

Чтобы не создавать новый объект Серии или новые элементы в Серии каждый раз, когда вы хотите интерполировать данные, используйте RedBlackPy . См. Пример кода ниже:

import redblackpy as rb

# we do not include missing data
index = [0,1,3,4,6]
data = [10,20,40,50,30]
# create Series object
series = rb.Series(index=index, values=data, dtype='float32',
                   interpolate='linear')

# Now you have access at any key using linear interpolation
# Interpolation does not creates new items in Series
print(series[2]) # prints 30
print(series[5]) # prints 40
# print Series and see that keys 2 and 5 do not exist in series
print(series)

Последний вывод следующий:

Series object Untitled
0: 10.0
1: 20.0
3: 40.0
4: 50.0
6: 30.0
питон, NumPy, интерполяция, линейно-интерполяции,
Похожие вопросы