Удаление подмассива из большего многомерного массива без изменения размеров - PullRequest
3 голосов
/ 09 мая 2020

Имея массивы a и b, я хотел бы получить массив c, исключающий a из b.

a=np.array([8,14])

[ 8 14]

b=np.array([[3,2],[8,10],[8,14],[17,65]])

[[ 3  2]
 [ 8 10]
 [ 8 14]
 [17 65]]

Желаемый c:

print(c)
[[ 3  2]
 [ 8 10]
 [17 65]]

numpy удаление, похоже, не работает должным образом, потому что он принимает индекс в качестве входных данных для удаления раздела массива.

np.delete(b, a)
[ 3  2  8 10 8 14 17 65]

1 Ответ

1 голос
/ 09 мая 2020

попробуйте это:

c = b[np.any(b != a, axis=(1))]
print(c)
...