python - 用于关联字典列表和列表列表的 pythonian 单行代码

Question

给定一个项目列表 [音乐家] 和一个词典列表 [乐器] 如何以 pythonic 方式在它们之间建立关联？

因此，为了简化我的真实场景，请考虑以下内容:

musicians  = [[700,"James","Hetfield", "jh@metallica.com","N/A"],
              [701,"Lars","Ulrich","lu@metallica.com","N/A"],
              [702,"Kirk","Hammett","kh@metallica.com","N/A"],
              [703,"Robert","Trujillo", "rt@metallica.com","N/A"]]

instruments= ({700:"guitar"},{701:"drums"})

我的目标是用 instruments in instruments 替换 N/A is musicians。

下面的代码可以解决问题(但直觉来自 java/c++)

for m in musicians:
    for i in instruments:
        if m[0] in i:
            m[4]=i[m[0]]

正确的预期结果是:

[[700, 'James', 'Hetfield', 'jh@metallica.com', 'guitar'], [701, 'Lars', 'Ulrich', 'lu@metallica.com', 'drums'], [702, 'Kirk', 'Hammett', 'kh@metallica.com', 'N/A'], [703, 'Robert', 'Trujillo', 'rt@metallica.com', 'N/A']]

问:python有经典的方法吗？

Answer 1

您可以使用 collections.ChainMap组合您的乐器映射，然后使用列表理解:

from collections import ChainMap

cm = ChainMap(*instruments)
musicians = [[*x[:-1], cm.get(x[0], x[-1])] for x in musicians]

print(musicians)

# [[700, 'James', 'Hetfield', 'jh@metallica.com', 'guitar'],
#  [701, 'Lars', 'Ulrich', 'lu@metallica.com', 'drums'],
#  [702, 'Kirk', 'Hammett', 'kh@metallica.com', 'N/A'],
#  [703, 'Robert', 'Trujillo', 'rt@metallica.com', 'N/A']]