Как извлечь индекс значения в массив numpy?

Question

Вот сценарий:

У меня есть следующие переменные:

val = [('am', '<f8'), ('fr', '<f8')] # val is type numpy.recarray  

am = [12.33, 1.22, 5.43, 15.23] # am is type numpy.ndarray 

fr = [0.11, 1.23, 2.01, 1.01] # fr is type numpy.ndarray 

Что мне нужно, чтобы обнаружить индекс для ат = 12,33 и я = 15.23, когда извлеченный (в этом случае индексы [ 0] и [3]), мне нужно создать новые переменные:

new_am = [12.33, 15.23] 

new_fr = [0.11, 1.01] 

Мой вопрос: Любая идея о том, как извлечь индексы?

Я уже использовал .index и np.where, но это, кажется, есть проблемы, так как я получил сообщение об ошибке для .index ---> «AttributeError: '' numpy.ndarray объект не имеет атрибута .index "и для np.where возвращение индекса ничего ---> "массив ([], DTYPE = int64)"

Спасибо за любую идею!

Answer 1

Вы можете в np.in1d, который возвращает массив булево, чтобы указать, если элемент находится в другом массиве:

import numpy as np 
am = np.array([12.33, 1.22, 5.43, 15.23]) 
fr = np.array([0.11, 1.23, 2.01, 1.01]) 

index = np.where(np.in1d(am, [12.33, 15.23]))[0] 
index 
# array([0, 3]) 

am[index] 
# array([ 12.33, 15.23]) 

fr[index] 
# array([ 0.11, 1.01]) 

---

Или, может быть, у вас есть массив с атрибутами:

new_arr = np.array(zip(am, fr), dtype=val) 

index = np.where(np.in1d(new_arr['am'], [12.33, 15.23]))[0] 

new_am = new_arr[index]['am'] 
new_fr = new_arr[index]['fr'] 

new_am 
# array([ 12.33, 15.23]) 

new_fr 
# array([ 0.11, 1.01])