使用Numpy数组作为查找表

一些示例数据：

import numpy as nplookup = np.array([[  1.     ,   3.14   ,   4.14   ],        [  2.     ,   2.71818,   3.7    ],        [  3.     ,  42.     ,  43.     ]])a = np.array([[ 1, 11],   [ 1, 12],   [ 2, 21],   [ 3, 31]])

在查找表中建立一个从键到行号的字典：

mapping = dict(zip(lookup[:,0], range(len(lookup))))

然后，您可以使用字典来匹配行。例如，如果您只想加入表：

>>> np.hstack((a, np.array([lookup[mapping[key],1:]       for key in a[:,0]])))array([[  1.     ,  11.     ,   3.14   ,   4.14   ],       [  1.     ,  12.     ,   3.14   ,   4.14   ],       [  2.     ,  21.     ,   2.71818,   3.7    ],       [  3.     ,  31.     ,  42.     ,  43.     ]])