python - numpy 获取所有元素都大于阈值的列索引

Question

我想找到一个 numpy 数组的列索引，其中列的所有元素都大于阈值。

例如，

 X = array([[ 0.16,  0.40,  0.61,  0.48,  0.20],
            [ 0.42,  0.79,  0.64,  0.54,  0.52],
            [ 0.64,  0.64,  0.24,  0.63,  0.43],
            [ 0.33,  0.54,  0.61,  0.43,  0.29],
            [ 0.25,  0.56,  0.42,  0.69,  0.62]])

在上面的例子中，如果阈值是0.4，我的结果应该是1,3。

Answer 1

您可以使用 np.where 与每列的 min 进行比较:

large = np.where(X.min(0) >= 0.4)[0]