Сравнение двух массивов numpy и элементов удаления

Question

Я проходил несколько решений, но я не могу найти решение, в котором я нуждаюсь.

У меня есть два numpy массивов. Возьмем здесь небольшой пример.

x = [1,2,3,4,5,6,7,8,9] 
y = [3,4,5] 

Я хочу сравнить x и y и удалить те значения x, которые находятся в y.

Так что я ожидаю, что моя final_x быть

final_x = [1,2,6,7,8,9] 

я узнал, что np.in1d возвращает булево массив той же длины, что x Правда, где элемент x в y и ложь в противном случае. Но как его использовать, если не какой-либо другой метод для получения моего final_x. ??

Answer 1

Просто передайте отрицанием версию булева массива, возвращенного np.in1d в массив x:

>>> x = np.array([1,2,3,4,5,6,7,8,9]) 
>>> y = [3,4,5] 
>>> x[~np.in1d(x, y)] 
array([1, 2, 6, 7, 8, 9]) 

Answer 2

Если у вас действительно есть Numpy массивы, то вы можете использовать numpy.setdiff1d ниже

import numpy as np 

x = np.array([1,2,3,4,5,6,7,8,9]) 
y = np.array([3,4,5]) 

z = np.setdiff1d(x, y) 
# array([1, 2, 6, 7, 8, 9]) 

Answer 3

Вы можете использовать встроенные наборы:

final_x = set(x) - set(y) 

и вычесть второй из первого. Вы можете конвертировать final_x в list или numpy.array, если вы чувствуете себя такой склонной.