如何使用字典翻译/替换数组的元素？

Question

我有一个numpy的阵列，其具有数百其是大写字母的元素，在没有特定的顺序

import numpy as np 
abc_array = np.array(['B', 'D', 'A', 'F', 'H', 'I', 'Z', 'J', ...]) 

在此numpy.ndarray每个元素是一个numpy.string_。

我也有一个“翻译词典”，以键/值对，这样的大写字母对应于一个城市

transdict = {'A': 'Adelaide', 'B': 'Bombay', 'C': 'Cologne',...} 

只有26个对字典中的transdict，但也有数以百计的信件在numpy数组中，我必须翻译。

什么是最有效的方法来做到这一点？

我考虑过使用numpy.core.defchararray.replace(a, old, new, count=None)[source]，但是这会返回ValueError，因为numpy数组的大小与字典键/值不同。

AttributeError: 'numpy.ndarray' object has no attribute 'translate'

Answer 1

请问这个怎么办？有时候，纯Python是处理这种事情的好方法。下面构建一个翻译列表（很容易转换回numpy数组）和连接的输出。

import numpy as np 
abc_array = np.array(['B', 'D', 'A', 'F', 'H', 'I', 'Z', 'J']) 

transdict = {'A': 'Adelaide', 
      'B': 'Bombay', 
      'C': 'Cologne', 
      'D': 'Dresden', 
      'E': 'Erlangen', 
      'F': 'Formosa', 
      'G': 'Gdansk', 
      'H': 'Hague', 
      'I': 'Inchon', 
      'J': 'Jakarta', 
      'Z': 'Zambia' 
} 

phoenetic = [transdict[letter] for letter in abc_array] 
print ' '.join(phoenetic) 

从这个输出是：

Bombay Dresden Adelaide Formosa Hague Inchon Zambia Jakarta 

Answer 2

随着蛮力NumPy broadcasting -

idx = np.nonzero(transdict.keys() == abc_array[:,None])[1] 
out = np.asarray(transdict.values())[idx] 

通过基于np.searchsorted搜索和索引 -

sort_idx = np.argsort(transdict.keys()) 
idx = np.searchsorted(transdict.keys(),abc_array,sorter = sort_idx) 
out = np.asarray(transdict.values())[sort_idx][idx] 

样品运行 -

In [1]: abc_array = np.array(['B', 'D', 'A', 'B', 'D', 'A', 'C']) 
    ...: transdict = {'A': 'Adelaide', 'B': 'Bombay', 'C': 'Cologne', 'D': 'Delhi'} 
    ...: 

In [2]: idx = np.nonzero(transdict.keys() == abc_array[:,None])[1] 
    ...: out = np.asarray(transdict.values())[idx] 
    ...: 

In [3]: out 
Out[3]: 
array(['Bombay', 'Delhi', 'Adelaide', 'Bombay', 'Delhi', 'Adelaide', 
     'Cologne'], 
     dtype='|S8') 

In [4]: sort_idx = np.argsort(transdict.keys()) 
    ...: idx = np.searchsorted(transdict.keys(),abc_array,sorter = sort_idx) 
    ...: out = np.asarray(transdict.values())[sort_idx][idx] 
    ...: 

In [5]: out 
Out[5]: 
array(['Bombay', 'Delhi', 'Adelaide', 'Bombay', 'Delhi', 'Adelaide', 
     'Cologne'], 
     dtype='|S8')