以元素方式比较计算不同元素的数量

时间:2017-10-07 18:26:44

标签: arrays numpy

假设我有两个NumPy数组:

A = [1, 1, 2, 3, 4, 8, 9]
B = [1, 2, 5, 3, 5, 8, 10]

是否有任何内置函数用于元素比较和不同元素的输出数量?在上面的示例中,输出4(索引:1,2,4,6是不同的)

1 个答案:

答案 0 :(得分:0)

没有单一功能(您需要两个),但您可以使用NumPy not_equalcount_nonzero

import numpy as np

print(np.count_nonzero(np.not_equal(A, B)))
# 4

这明显快于np.not_equal(A, B).sum()