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

j2qf4p5b 于 2021-09-08 发布在 Java

关注(0)|答案(1)|浏览(348)

有一个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]]]

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的行转换为0 < 或 = 到 3 . 如果需要，请更改条件。