如何将numpy数组中的id值更改为带有字典的字符串numpy数组

Question

我正在研究将学校名称的ID转换为以numpy数组结构的实际学校名称的脚本。

例如

[[1,2,3],[3,6,7]]

变

[[school-a,school-b,school-c],[school-c,school-f,school-g]

学校和id一起坐在python字典中。

我尝试这样做：

for x in np.nditer(finalarray, op_flags=['readwrite']):
    x[...] = school_ids.get(int(x))
    print(school_ids.get(int(x)))
print(finalarray)

但这给出了错误：

ValueError: invalid literal for int() with base 10: 'school-a'

保持numpy数组的结构不变是很重要的，因为我还想到了仅迭代每个项目，但是随后结构丢失了。

Answer 1

使用这篇文章的解决方案：

x = np.array([[1,1,3], [2,2,2]])
d = {1: 'a', 2:'b', 3:'c'}
np.vectorize(d.get)(x)
>> array([['a', 'a', 'c'],
   ['b', 'b', 'b']], dtype=object)

Answer 2

假设我有一个命令：

dictt = {
    0: 'school-a',
    1: 'school-b',
    2: 'school-c',
    3: 'school-d',
    4: 'school-e',
    5: 'school-f',
    6: 'school-g',
    7: 'school-h',
    8: 'school-i'
}

_ids = np.array([[1,2,3],[3,6,7]])
school_ids = np.array(list(dictt.values()))
print school_ids[_ids-1]

得到：

[['school-a' 'school-b' 'school-c']
 ['school-c' 'school-f' 'school-g']]