如何根据结尾用数字替换列表中的字符串？

Question

假设我有一个这样的列表

myList = ['A_x1', 'B_x2', 'C_x1', 'D_x3', 'E_x1']

还有这样的字典

myDict = {'_x1': 0.1, '_x2': 0.5, '_x3': 0.7}

现在我想将myList 中的所有字符串替换为myDict 中存储的与该字符串结尾相关联的值。我当然可以通过以下方式实现：

for ind, l in enumerate(myList):
    for k in myDict.iterkeys():
        if l.endswith(k):
            myList[ind] = myDict[k]

这给了我想要的输出：

[0.1, 0.5, 0.1, 0.7, 0.1].

但是如何使用例如map 或其他更有效的方法？

Answer 1

for i, e in enumerate(myList):
    key = e[e.index("_"):]
    if key not in myDict: continue
    myList[i] = myDict[key]

Answer 2

根据您的真实数据的外观，可以使用以下内容：

myList = ['A_x1', 'B_x2', 'C_x1', 'D_x3', 'E_x1']
myDict = {'_x1': 0.1, '_x2': 0.5, '_x3': 0.7}

print [myDict[x[-3:]] for x in myList]

给予：

[0.1, 0.5, 0.1, 0.7, 0.1]

Answer 3

如果所有字典键始终以 _（下划线）开头，您可以使用 str.rsplit 和列表理解 -

myList[:] = [myDict.get('_' + elem.rsplit('_',1)[1],elem) for elem in myList]

演示 -

>>> myList = ['A_x1', 'B_x2', 'C_x1', 'D_x3', 'E_x1']
>>> myDict = {'_x1': 0.1, '_x2': 0.5, '_x3': 0.7}
>>> myList[:] = [myDict.get('_' + elem.rsplit('_',1)[1],elem) for elem in myList]
>>> myList
[0.1, 0.5, 0.1, 0.7, 0.1]

---

对于一般版本（基本上是您在使用next() 完成的列表理解中的嵌套循环）。示例 -

myList[:] = [myDict.get(next((key for key in myDict if elem.endswith(key)),None),elem) 
             for elem in myList]

演示 -

>>> myList = ['A_x1', 'B_x2', 'C_x1', 'D_x3', 'E_x1']
>>> myDict = {'_x1': 0.1, '_x2': 0.5, '_x3': 0.7}
>>>
>>> myList[:] = [myDict.get(next((key for key in myDict if elem.endswith(key)),None),elem)
...              for elem in myList]
>>>
>>> myList
[0.1, 0.5, 0.1, 0.7, 0.1]

但为了可读性，我更喜欢嵌套的 for 循环（尽管它可能会慢一点）。

Answer 4

# first map through lambda gets the endings.
# second map applies the endings as keys to myDict

print(list(map(myDict.get,map(lambda x: x[-3:],myList))))

[0.1, 0.5, 0.1, 0.7, 0.1]