猿问

从 numpy 数组 a 中删除数组 b 中存在的所有项目 - Python

有一个更好的方法吗?我使用了 np.setdiff1D ,我期望的输出是 5。如果 [1, 2, 3, 4] 是正确答案。有人可以详细解释为什么吗?谢谢!**


import numpy as np


a = np.array([1, 2, 3, 4, 5])

b = np.array([5, 6, 7, 8, 9])


result = np.setdiff1d(a, b)

print(result)


# Output

[1 2 3 4]


Cats萌萌
浏览 144回答 1
1回答

潇潇雨雨

根据文档,返回 中不存在的setdiff1d(ar1, ar2)唯一值。 https://numpy.org/doc/stable/reference/ generated/numpy.setdiff1d.htmlar1ar2[1, 2, 3, 4]a是 中 中不存在的唯一值b。这就实现了帖子标题所要求的:5是 中唯一的唯一值,a也是 中的b,因此这是操作后唯一不存在的值。
随时随地看视频慕课网APP

相关分类

Python
我要回答