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

前端 未结 5 811
一个人的身影
一个人的身影 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 06:19

    Using np.where:

    >>> t.T[np.where(t[1])].T
    array([[ 1,  0,  5,  7,  8],
           [11, 13, 15, 17, 18]])
    

提交回复
热议问题