用字典中的相应值替换2D列表中的元素？

Question

我有一个二维Python列表。 我也有字典。 我想生成一个与列表形状相同的numpy矩阵，其中列表中的每个元素都替换为字典中的相应值。

例如，这是我的Python列表：

[[John Mary]

[Jim Jane]

我想以此创建实例（一个numpy矩阵，而不是一个二维Python列表）：

[[85 53]

[76 45]

从字典中检索值的位置。

最快的方法是什么？ 我需要这个非常快。

Answer 1

您可以将函数应用于整个numpy数组：

>>> pylist = [['John','Mary'],['Jim', 'Jane']]
>>> nparray = np.asarray(pylist)
>>> print nparray
[['John' 'Mary']
 ['Jim' 'Jane']]
>>> dt = {'John':86, 'Mary':53, 'Jim':76, 'Jane':45}
>>> func = np.vectorize(lambda x: dt[x])
>>> print func(nparray)
[[86 53]
 [76 45]]