Эта проблема кажется легкой, но я не могу получить красивое решение. У меня есть два массива numpy (A и B), и я хочу получить индексы A, где элементы A находятся в B, а также получить индексы A, где элементы не находятся в B.
Итак, если
A = np.array([1,2,3,4,5,6,7])
B = np.array([2,4,6])
В настоящее время я использую
C = np.searchsorted(A,B)
который использует тот факт, что A
по порядку, и дает мне [1, 3, 5]
, индексы элементов, которые находятся в A
. Это здорово, но как мне получить D = [0,2,4,6]
, индексы элементов A
, которые не находятся в B
?