Как применить не целое число -> integer dict к массиву numpy?

Question

Скажем, у меня есть эти два массива:

dictionary = np.array(['a', 'b', 'c']) 
array = np.array([['a', 'a', 'c'], ['b', 'b', 'c']]) 

И я хотел бы заменить каждый элемент в array с индексом его значения в dictionary. Итак:

for index, value in enumerate(dictionary): 
    array[array == value] = index 
array = array.astype(int) 

Чтобы получить:

array([[0, 0, 2], 
     [1, 1, 2]]) 

Есть ли Векторизованный способ сделать это? Я знаю, что если array уже содержит индексы, и мне нужны строки в dictionary, я мог бы просто сделать dictionary[array]. Но мне действительно нужен «поиск» строк.

(я также вижу this answer, но интересно, если что-то новое было доступно с 2010 года)

Answer 1

Если ваш словарь сортируется, и словарь и массив содержит те же элементы, np.unique делает трюк

uniq, inv = np.unique(array, return_inverse=True) 
result = inv.reshape(array.shape) 

Если некоторые элементы отсутствуют в массиве:

uniq, inv = np.unique(np.r_[dictionary, array.ravel()], return_inverse=True) 
result = inv[len(dictionary):].reshape(array.shape) 

Общий случай:

uniq, inv = np.unique(np.r_[dictionary, array.ravel()], return_inverse=True) 
back = np.empty_like(inv[:len(dictionary)]) 
back[inv[:len(dictionary)]] = np.arange(len(dictionary)) 
result=back[inv[len(dictionary):]].reshape(array.shape) 

Объяснение: np.unique в виде мы используем его здесь возвращает уникальные элементы в отсортированном порядке и индексы в этот отсортированный список каждого элемента аргумента. Итак, чтобы получить индексы в исходном словаре, нам нужно переназначить индексы. Мы знаем, что uniq[inv[:len(uniq)]] == dictionary. Поэтому мы должны решить X[inv[:len(uniq)]] == np.arange(len(uniq)), что и делает код.