如何將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']]