如何计算Dataframe中两列中具有相同值的行

时间:2021-05-05 10:10:48

标签: python pandas dataframe numpy

我正在做一个数据集研究,我得到了一个分类结果数据集,如下所示。

   Actual Prediction
0     DS    DS
1     DS    DS
2     WS    DS
3     WD    WD
4     WS    WS

我想比较 ActualPrediction,并且只返回具有 Actual == DSPrediction == DS 的行。

Desired Output:
    Actual Prediction
0    DS    DS
1    DS    DS
2    DS    DS

带有 Actual != DSPrediction == DS 的行被认为是错误分类

这样我就可以计算成功分类的准确性了。

我已经搜索了很多,但我无法通过尝试大量数据帧内置函数(例如计数、重复等)来解决这个问题。

非常感谢任何帮助!

1 个答案:

答案 0 :(得分:0)

B9

结果:

data = {'Actual': ['DS', 'WD'], 'Prediction':  ['DS', 'WD']}  
df = pd.DataFrame(data)
df[(df.Actual == "DS") & (df.Prediction == "DS")]