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

前端 未结 5 807
一个人的身影
一个人的身影 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条回答
  •  猫巷女王i
    2020-12-10 06:16

    Similar to Juh_, but more expressive, and avoiding some minor unnecessary performance overhead. A grand total of 12 highly pythonic, explicit and unambigious characters. This is really numpy 101; if you are still trying to wrap your head around this, you would do yourself a favor by reading a numpy primer.

    import numpy as np
    a = np.array([[ 0,  1,  2,  0,  4,  5,  6,  7,  8,  9],
                  [ 0, 11,  0, 13,  0, 15,  0, 17, 18,  0]])
    print a[:,a[1]!=0]
    

提交回复
热议问题