如何从 Python 中的二进制值对计算 TP/TN/FP/FN？

Question

我想编写一个 Python 函数，它在二进制值输入对（真值，预测）上根据输入返回真阳性/真阴性/假阳性/假阴性值。 到目前为止，我已经达到了所需的输出：

def func(truth, prediction):
    
    if prediction == 1:
        
        if truth == 1:
            return "TP"
        else:
            return "FP"

    elif truth == 1:
        return "FN"

    else:
        return "TN"

然而，这似乎是一个笨拙的解决方案，有没有更短、更优雅的方法？

（输入对应该是二进制整数 0/1）

Answer 1

Johnny Mopp的评论非常酷（虽然我认为顺序应该是['TN', 'FN', 'FP', 'TP'] ，但如果我在代码中遇到它，我必须三思而后行。 （一方面我可以数出我在生产代码中看到移位操作的次数。）

在 Python 3.10 中有一种处理此类事情的新方法： 结构模式匹配。 现在，这是我第一次尝试使用这个新功能，但它看起来是这样的：

def get_result(true, pred):
    """Decide if TP, FP, TN or FN"""
    match [true, pred]:
        case [1, 1]: return 'TP'
        case [1, 0]: return 'FN'
        case [0, 0]: return 'TN'
        case [0, 1]: return 'FP'

似乎工作：

>>> y = [1, 0, 1, 0]  # TP, FP, FN, TN
>>> ŷ = [1, 1, 0, 0]

>>> for yi, ŷi in zip(y, ŷ):
>>>     print(get_result(yi, ŷi))
TP
FP
FN
TN

如果您需要使用 Python 3.9 或更低版本，那么您可以通过以下方式压缩您的方法：

def get_result(true, pred):
    """Decide if TP, FP, TN or FN"""
    if pred:
        return 'TP' if true else 'FP'
    return 'FN' if true else 'TN'