具有数组a和b,我想获得从b中排除所有a的数组c。
a=np.array([8,14])
[ 8 14]
b=np.array([[3,2],[8,14],[17,65]])
[[ 3 2]
[ 8 14]
[17 65]]
所需的c是:
print(c)
[[ 3 2]
[17 65]]
numpy delete似乎没有按预期方式工作,因为它将索引作为删除数组部分的输入。
np.delete(b, a)
[ 3 2 8 14 17 65]
这是一个非笨拙的解决方案,但仍然可以在这里解决您的问题:
c = [i for i in b if i not in a]
print(c)