0

我有两个numpy大小相等的数组。它们包含值1、0和-1。我可以计算匹配元素和负元素的数量，但我不确定如何计算具有相同索引和零值的匹配元素。

我对如何在这里进行有点困惑。

这是一些代码：

print(actual_direction.shape)
print(predicted_direction.shape)
act = actual_direction
pre = predicted_direction
part1 = act[pre == 1]
part2 = part1[part1 == 1]
result1 = part2.sum()
part3 = act[pre == -1]
part4 = part3[part3 == -1]
result2 = part4.sum() * -1
non_zeros = result1 + result2
zeros = len(act) - non_zeros
print(f'zeros : {zeros}\n')
print(f'non_zeros : {non_zeros}\n')
final_result = non_zeros + zeros
print(f'result1 : {result1}\n')
print(f'result2 : {result2}\n')
print(f'final_result : {final_result}\n')

这是打印输出：

(11279,)
(11279,)
zeros : 5745.0
non_zeros : 5534.0
result1 : 2217.0
result2 : 3317.0
final_result : 11279.0

所以我在这里所做的只是从数组的总长度中减去一个和负数的总和。我不能假设差异（零：5745）包含所有包含零的匹配元素，我可以吗？

4

3 回答 3

3

你可以试试这个：

import numpy as np

a=np.array([1,0,0,1,-1,-1,0,0])
b=np.array([1,0,0,1,-1,-1,0,1])
summ = np.sum((a==0) & (b==0))
print(summ)

输出：

3
于 2020-07-05T19:32:03.820 回答
1

您可以numpy.ravel()用来展平数组，然后用来zip()并排比较每个元素：

import numpy as np

ar1 = np.array([[1, 0, 0],
                [0, 1, 1],
                [0, 1, 0]])

ar2 = np.array([[0, 0, 0],
                [1, 0, 1],
                [0, 1, 0]])

count = 0
for e1, e2 in zip(ar1.ravel(), ar2.ravel()):
    if e1 == e2:
        count += 1

print(count)

输出：

6

您也可以这样做来列出找到的所有匹配项，并打印出金额：

dup = [e1 for e1, e2 in zip(ar1.ravel(), ar2.ravel()) if e1 == e2]
print(dup)
print(len(dup))

输出：

[0, 0, 1, 0, 1, 0]
6
于 2020-07-05T19:32:20.733 回答
0

您有两个数组，并且想要计算这两个数组都为 0 的位置，对吗？

您可以检查数组在哪里满足您的要求(a == 0)，然后使用“和”运算符&检查两个数组在哪里满足您的要求：

import numpy as np
a = np.array([1, 0, -1, 0, -1, 1, 1, 1, 1])
b = np.array([1, 0, -1, 1, 0, -1, 1, 0, 1])

both_zero = (a == 0) & (b == 0)  # [False,  True, False, False, False, False]

both_zero.sum()  # 1

在您更新的问题中，您似乎对实际值和预测之间的异同感兴趣。为此，混淆矩阵非常适合。

from sklearn.metrics import confusion_matrix
confusion_matrix(a, b, labels=[-1, 0, 1])

会给你一个混淆矩阵作为输出，告诉你有多少个 -1 被预测为 -1、0 和 1，对于 0 和 +1 也是如此：

[[1 1 0]   # -1s predicted as -1, 0 and 1
 [0 1 1]   # 0s predicted as -1, 0 and 1
 [1 1 3]]  # 1s predicted as -1, 0 and 1
于 2020-07-05T19:35:23.543 回答