如何基于值比较python中的两个数组

Question

我在做分类.我有两个数组,第一个是'Actual',第二个是'Predicted'.我想比较这两个数组.假设第一个数组是:

Actual = [1, 1, 2, 3, 1]

这告诉我们第一个,第二个和最后一个索引对应于类1.

'Predicted'数组是:

Predicted = [1, 1, 1, 3, 3]

这告诉我们第一和第二个索引已经准确预测.

我希望输出告诉我们那些准确预测为1的索引,如下所示:

output = [True, True, False, False, False]

更新 我想仅根据值1进行评估.如果你看到,第四个预测值是由3准确预测的,但我不希望这样,因为我想要评估1值.

Answer 1

假设length两个列表相同:

>>> [(x == y == 1) for x, y in zip(Actual, Predicted)]
[True, True, False, False, False]

感到安全;

>>> from itertools import izip_longest
>>> [(x == y == 1) for x, y in izip_longest(Actual, Predicted, fillvalue=0)]
[True, True, False, False, False]