使用条件

Question

截至目前，我有一个numpy数组，看起来像：

A =  [  [5, 8, 6, 2], 
        [5, 8, 6, 2], 
        [...], ...       ]

比方说，我还有另一个1D numpy数组，看起来像数组B，但其值对应于A内部的每个列表。

B = [0.4, 0.6, 0.3, ...]

现在，我要删除列表中B中具有大于0.5的对应值的任何列表（这可以由名为thresh的变量表示）。结果数组应该看起来像（因为第二个列表的值更大）：

C =  [  [5, 8, 6, 2], 
        [...], ...       ]

我正在寻找一种基于numpy / pythonic的良好方式来实现这一目标。我知道np.delete，但不了解如何在有条件的情况下使用它。

Answer 1

条件set切片

setlocal EnableDelayedExpansion