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

后端 未结 5 1822
野的像风
野的像风 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:24

    check if not nan then reverse order of columns and take argmax then subtract from number of columns

    a.shape[1] - (~np.isnan(a))[:, ::-1].argmax(1) - 1
    
    array([3, 2, 6, 3, 0, 3])
    

提交回复
热议问题