numpy - 比较两个数组的每个元素并返回矩阵

时间:2018-02-09 17:50:47

标签: python numpy matrix vector machine-learning

我有预测概率的向量。我想将每个概率与几个阈值进行比较,最后得到矩阵结果[i,j],如果概率[i]大于阈值[j],则保存布尔值。

这就是我到目前为止......

y = np.random.randint(2, size=6)

pred = np.array([0.21,0.37,0.18,0.88,0.54,0.23])

thresh = np.array([0.25, 0.5, 0.75])

pred_true = []
for i in range(thresh.shape[0]): #for each threshold
    pred_true.append(pred > thresh[i])

v_pred_true = np.vstack(pred_true).T
v_pred_true

给出:

Out[40]: 
array([[False, False, False],
       [ True, False, False],
       [False, False, False],
       [ True,  True,  True],
       [ True,  True, False],
       [False, False, False]], dtype=bool)

但是我想找到一种比较简单的方法来做到这一点。

0 个答案:

没有答案