python - 如何将数组中的值替换为 Python 中列表中的值？

Question

我有一个像这样的数组(numpy.ndarray 类型)

arr=array([[1, 5, 1],
          [4, 2, 0]])

和值列表:

values=['father','mother','sister','brother','aunt','uncle']

我想用列表 values 中的项目替换数组 arr 中的数字，使用数组的项目作为列表值的索引:arr[0,0 ]=值[arr[0,0]]

这是我想要的例子

arr=array([['mother', 'uncle', 'mother'],
           ['aunt', 'sister', 'father']])

有什么优雅的 pythonic 方法可以做到这一点吗？

感谢您提前提供帮助 =)

Answer 1

您可以将 values 转换为 numpy 数组，然后使用简单的索引:

>>> values = np.array(values)
>>> 
>>> values[arr]
array([['mother', 'uncle', 'mother'],
       ['aunt', 'sister', 'father']], 
      dtype='|S7')

阅读有关索引的更多信息:http://docs.scipy.org/doc/numpy-1.10.0/reference/arrays.indexing.html