Python数据框获取每行最后一个非空列的值

时间:2020-02-26 06:15:41

标签: python pandas dataframe

我有一个如下数据框:

    ID  2016    2017    2018    2019    2020
0   1   1.64    NaN     NaN     NaN     NaN
1   2   NaN     NaN     NaN     0.78    NaN
2   3   1.11    0.97    1.73    1.23    0.87
3   4   0.84    0.74    1.64    1.47    0.41
4   5   0.75    1.05    NaN     NaN     NaN

我想从最后一个非null列中获取值,例如:

    ID  2016    2017    2018    2019    2020    LastValue
0   1   1.64    NaN     NaN     NaN     NaN     1.64
1   2   NaN     NaN     NaN     0.78    NaN     0.78
2   3   1.11    0.97    1.73    1.23    0.87    0.87
3   4   0.84    0.74    1.64    1.47    0.41    0.41
4   5   0.75    1.05    NaN     NaN     NaN     1.05

我尝试按如下所示反向循环浏览年列,但无法完全实现我想要的。

for i in reversed(df.columns[1:]):
     if df[i] is not None:
          val = df[i]

您能帮忙解决这个问题吗?谢谢。

1 个答案:

答案 0 :(得分:1)

想法是先选择所有列,而不用DataFrame.iloc,然后向前填充缺少值的行,最后选择最后一列:

df['LastValue'] = df.iloc[:, 1:].ffill(axis=1).iloc[:, -1]
print (df)
   ID  2016  2017  2018  2019  2020  LastValue
0   1  1.64   NaN   NaN   NaN   NaN       1.64
1   2   NaN   NaN   NaN  0.78   NaN       0.78
2   3  1.11  0.97  1.73  1.23  0.87       0.87
3   4  0.84  0.74  1.64  1.47  0.41       0.41
4   5  0.75  1.05   NaN   NaN   NaN       1.05

详细信息

print (df.iloc[:, 1:].ffill(axis=1))
   2016  2017  2018  2019  2020
0  1.64  1.64  1.64  1.64  1.64
1   NaN   NaN   NaN  0.78  0.78
2  1.11  0.97  1.73  1.23  0.87
3  0.84  0.74  1.64  1.47  0.41
4  0.75  1.05  1.05  1.05  1.05