python - 根据第一个元素的状态清除行

Question

有一个 3d 数组:

输入:

    [[[0,2,3,4]
    [4,2,3,4]
    [6,2,3,4]]
    [[2,2,3,4]
    [3,2,3,4]
    [2,2,3,4]]]

如何使 numpy 数组看起来像这样？

rule: if array[:,:,0] < 3

输出:

    [[[0,0,0,0]
    [4,2,3,4]
    [6,2,3,4]]
    [[0,0,0,0]
    [0,0,0,0]
    [0,0,0,0]]]

Answer 1

这是一种方法:

a[a[:,: ,0] <= 3, :] = 0

输出:

array([[[0, 0, 0, 0],
        [4, 2, 3, 4],
        [6, 2, 3, 4]],

       [[0, 0, 0, 0],
        [0, 0, 0, 0],
        [0, 0, 0, 0]]])

注意:我假设您想将值为 < 的行转为 0或=至3 。如果需要，更改条件。