删除阵列中位于另一个阵列中的所有数字

Question

我有一个数组"可移动"包含来自另一个数组"All"的几个数字,包含从0到k的所有数字.

我想删除可移动中列出的所有数字.

All = np.arange(k)
removable = np.ndarray([1, 3, 4 , 7, 9, ..., 200])

for i in removable:
    if i in All:
        All.remove(i)

ndarray没有删除属性,但我确定在numpy中有一个简单的方法来解决这个问题,但我在纪录片中找不到它.

Answer 1

您可以使用NumPy中的函数setdiff1d:

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