Python：在多个 np.arrays 上映射字典

Question

如何将字典映射到 np.ndarrays 列表上？

我有一本包含一系列键和值的字典。它看起来像这样：

{1: 0.5,
 2: 0.51,
 3: 0.34,
 4: 0.38,
 5: 0.4,
 6: 0.27,}

此外，我有一个列表对象，其中包含一系列 numpy 数组，如下所示：

[array([1,2,3]),
 array([4,3,5,6]),
 array([1,4,6,2,3])]

我想将字典映射到数组上，以用字典中对应的键值替换每个数组中的每个值。它会产生这样的结果：

[array([0.5,0.51,0.34]),
 array([0.38,0.34,0.4,0.27]),
 array([0.5,0.38,0.27,0.51,0.34])]

最后，我想取上述结构中每个数组的平均值，并将这个值附加到数据框中。

有什么建议吗？

Answer 1

这是一种解决方案。

import numpy as np

d = {1: 0.5,
 2: 0.51,
 3: 0.34,
 4: 0.38,
 5: 0.4,
 6: 0.27,}

lst = [np.array([1,2,3]),
 np.array([4,3,5,6]),
 np.array([1,4,6,2,3])]

lst2 = list(map(np.vectorize(d.get), lst))

# [array([ 0.5 ,  0.51,  0.34]),
#  array([ 0.38,  0.34,  0.4 ,  0.27]),
#  array([ 0.5 ,  0.38,  0.27,  0.51,  0.34])]