Delete element from multi-dimensional numpy array by value

前端 未结 5 1717
抹茶落季
抹茶落季 2020-12-18 10:06

Given a numpy array

a = np.array([[0, -1, 0], [1, 0, 0], [1, 0, -1]])

what\'s the fastest way to delete all elements of value -1

5条回答
  •  -上瘾入骨i
    2020-12-18 10:55

    How about this?

    print([[y for y in x if y > -1] for x in a])
    [[0, 0], [1, 0, 0], [1, 0]]
    

提交回复
热议问题