比较2个数组的数目

Question

如何编写一个函数来比较a和b，然后输出在a中但不在b中的数组？

import numpy as np
a = np.array([1,2,3,6,80,4,5,6])
b = np.array([1,2,3,70,50])

预期结果

[80,4,5,6]

Answer 1

您可以使用setxor1d。：

c = np.setxor1d(a, b)