Getting the last non-nan index of a sorted numpy matrix or pandas dataframe

后端 未结 5 1807
野的像风
野的像风 2020-12-21 04:50

Given a numpy array (or pandas dataframe) like this:

import numpy as np

a = np.array([
[1,      1,      1,    0.5, np.nan, np.nan, np.nan],
[1,      1,              


        
5条回答
  •  轮回少年
    2020-12-21 05:36

    This solution doesn't require the array to be sorted. It just returns the last non nan item along axis 1.

    (~np.isnan(a)).cumsum(1).argmax(1)
    

提交回复
热议问题