小编Jos*_*Lim的帖子

我有一个形状为 [6,2] 的二维 numpy 数组，我想删除第三个元素包含 0 的子数组。

array([[0, 2, 1], #Input
       [0, 1, 1],
       [1, 1, 0],
       [1, 0, 2],
       [0, 2, 0],
       [2, 1, 2]])

array([[0, 2, 1], #Output
       [0, 1, 1],
       [1, 0, 2],
       [2, 1, 2]]) 

我的代码是 positives = gt_boxes[np.where(gt_boxes[range(gt_boxes.shape[0]),2] != 0)]

它有效，但有没有一种简化的方法？