Я пытаюсь разместить записи в numpy.recarray на основе общих значений одного из полей recarrays и внешнего массива. Например,
a = np.array([(10, 'Bob', 145.7), (20, 'Sue', 112.3), (10, 'Jim', 130.5)],
dtype=[('id', 'i4'), ('name', 'S10'), ('weight', 'f8')])
a = a.view(np.recarray)
b = np.array([10,30])
Я хочу взять пересечение a.id и b, чтобы определить, какие записи извлечь из повторного массива, чтобы я вернулся:
(10, «Боб», 145,7)
(10, «Джим», 130,5)
Наивно я пробовал:
common = np.intersect1d(a.id, b)
subset = a[common]
но, конечно, это не работает, потому что нет [10]. Я также попытался сделать это путем создания обратного диктанта между полем id и индексом и оттуда подмножеством, например
id_x_index = {}
ids = a.id
indexes = np.arange(a.size)
for (id, index) in zip(ids, indexes):
id_x_index[id] = index
subset_indexes = np.sort([id_x_index[x] for x in ids if x in b])
print a[subset_indexes]
но тогда я переопределяю значения dict в id_x_index, если a.id имеет дубликаты, как в этом случае я получаю
(10, «Джим», 130,5)
(10, «Джим», 130,5)
Я знаю, что пропускаю какой-то простой способ получить соответствующие индексы в массиве данных. Спасибо за помощь.