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

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

    With numpy.delete:

    a = np.array([[0, 1, 2, 0, 4, 5, 6, 7, 8, 9], [0, 11, 0, 13, 0, 15, 0, 17, 18, 0]])
    
    indices = [i for (i,v) in enumerate(a[1]) if v==0]
    # [0, 2, 4, 6, 9]
    
    a = np.delete(a, indices, 1)
    # array([[ 1,  0,  5,  7,  8], [11, 13, 15, 17, 18]])
    

提交回复
热议问题