Numpy数组减法

时间:2017-06-21 16:53:34

标签: python arrays sorting numpy

我如何比较两个不同的numpy数组来创建第三个数组?我想写一个循环,通过两个数组并打印一个新的数组“c”,只有不在a或b中的值

例如说我有

a = [1,2,3,4]
b = [1,2,3,4,5,6,7,8,9]

我想要打印:

c = [5,6,7,8,9]

1 个答案:

答案 0 :(得分:2)

您可以在Numpy中使用set difference操作:numpy.setdiff1d。 来自Numpy docs

>>> a = np.array([1, 2, 3, 2, 4, 1])
>>> b = np.array([3, 4, 5, 6])
>>> np.setdiff1d(a, b)
array([1, 2])