python - 以特定方式替换数组中的 NaN

Question

我有一个形式的数组(此处简化):[1,NaN,NaN,7,NaN,27]。我想用已知值之间等距的值替换 NaN 的，因此上面的数组将变为 [1,3,5,7,17​​,27]。有没有一种快速的方法可以做到这一点(不使用一些 for 循环)？谢谢!

Answer 1

Pandas dataframe.interpolate() 函数基本上用于填充数据帧或系列中的 NA 值

import pandas as pd
import numpy as np

arr = [ 1, np.NaN, np.NaN, 7, np.NaN, 27]
//converting array in series 
print(pd.Series(arr).interpolate(method = 'linear', limit_direction = 'forward'))

参数
method = 'linear':忽略索引并将值视为等距。
limit_direction: {‘forward’, ‘backward’, ‘both’}，默认‘forward’如果指定了limit，则连续的NaN将往该方向填充。
限制: int，可选要填充的最大连续 NaN 数。必须大于 0。

print(pd.Series(arr).interpolate(method = 'linear', limit_direction = 'forward', limit = 1))
#5 won't  get printed
print(pd.Series(arr).interpolate(method = 'linear', limit_direction = 'backward', limit = 1))
#3 won't get printed

您可以根据您的要求尝试不同的变体。