Удалить определенные индексы в каждой строке numpy ndarray ⇐ Python

[Anonymous]

У меня есть целочисленные массивы типа:

Код: Выделить всё

import numpy as np

seed_idx = np.asarray([[0, 1],
[1, 2],
[2, 3],
[3, 4]], dtype=np.int_)

target_idx = np.asarray([[2,9,4,1,8],
[9,7,6,2,4],
[1,0,0,4,9],
[7,1,2,3,8]], dtype=np.int_)

Для каждой строки target_idx я хочу выбрать элементы, индексы которых не совпадают с индексами вseed_idx. Таким образом, результирующий массив должен иметь следующий вид:

Код: Выделить всё

[[4,1,8],
[9,2,4],
[1,0,9],
[7,1,2]]

Другими словами, я хочу сделать что-то похожее на np.take_along_axis(target_idx,seed_idx, axis=1), но исключив индексы вместо их сохранения.
Другими словами, я хочу сделать что-то похожее на np.take_along_axis(target_idx,seed_idx, axis=1), но исключив индексы вместо их сохранения.
p>
Какой самый элегантный способ сделать это? Меня на удивление раздражает найти что-то аккуратное.

Подробнее здесь: https://stackoverflow.com/questions/791 ... py-ndarray