Delete columns from numpy array depending on a condition on a single cell

前端 未结 5 799
一个人的身影
一个人的身影 2020-12-10 05:43

Above all, sorry for my bad English.

I have this array t:

array([[ 0,  1,  2,  0,  4,  5,  6,  7,  8,  9],
       [ 0, 11,  0, 13,  0,          


        
5条回答
  •  北海茫月
    2020-12-10 05:55

    I got this working like this:

    data = array([[ 0, 1, 2, 0, 4, 5, 6, 7, 8, 9], [ 0, 11, 0, 13, 0, 15, 0, 17, 18, 0]])
    res = array([(a, b,) for a, b in zip(data[0], data[1]) if b]).transpose()
    

    got the result

    In [23]: res
    Out[23]: 
    array([[ 1,  0,  5,  7,  8],
           [11, 13, 15, 17, 18]])
    

提交回复
热议问题