Numpy: Drop rows with all nan or 0 values

前端 未结 5 1361
梦毁少年i
梦毁少年i 2020-12-29 06:41

I\'d like to drop all values from a table if the rows = nan or 0.

I know there\'s a way to do this using pandas i.e pandas.dropna(how

5条回答
  •  情深已故
    2020-12-29 07:21

    import numpy as np
    
    a = np.array([
        [1, 0, 0],
        [0, np.nan, 0],
        [0, 0, 0],
        [np.nan, np.nan, np.nan],
        [2, 3, 4]
    ])
    
    mask = np.all(np.isnan(a) | np.equal(a, 0), axis=1)
    a[~mask]
    

提交回复
热议问题