python - 将 NaN 值插入 DataFrame 作为环

Question

我需要在 Dataframe 上插入 NaN 值，但我希望该插值能够获取 DataFrame 的第一个值，以防NaN 值是最后一个值。这是一个例子:

import pandas as pd
import numpy as np
df = pd.DataFrame.from_dict({"a": [1,2,3], "b":[1,2,np.nan]})

所以 DataFrame 是:

   a    b
0  1  1.0
1  2  2.0
2  3  NaN

但是当我像这样插入 nan 值时:

df.interpolate(method="linear", inplace=True)

我得到了:

   a    b
0  1  1.0
1  2  2.0
2  3  2.0

插值不使用第一个值来完成。由于该圆弧插值，我想要的输出将填充 1.5 的值。

Answer 1

一个可能的解决方案是添加第一行，插入并删除最后一行:

df = df.append(df.iloc[0]).interpolate(method="linear").iloc[:-1]
print (df)
     a    b
0  1.0  1.0
1  2.0  2.0
2  3.0  1.5

编辑:

更通用的解决方案:

df = pd.DataFrame.from_dict({"a": [1,2,3,4], "b":[np.nan,1,2,np.nan]})

df = pd.concat([df] * 3).interpolate(method="linear").iloc[len(df):-len(df)]

print (df)
   a         b
0  1  1.333333
1  2  1.000000
2  3  2.000000
3  4  1.666667

或者如果只需要处理最后的非缺失值:

df = pd.DataFrame.from_dict({"a": [1,2,3,4], "b":[np.nan,1,2,np.nan]})

df1 = df.ffill().iloc[[-1]]
df2 = df.bfill().iloc[[0]]
df = pd.concat([df1, df, df2]).interpolate(method="linear").iloc[1:-1]

print (df)
   a    b
0  1  1.5
1  2  1.0
2  3  2.0
3  4  1.5