比较2个数组的数量

问题描述

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

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

预期结果

[80,6]

解决方法

您可以使用 setxor1d。：

c = np.setxor1d(a,b)