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



给定一个项目列表[音乐家]和一个词典列表[intrumets],如何以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"})

我的目标是用乐器中的乐器取代N/A是音乐家。

下面的代码可以解决问题(但直觉来自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 中做到这一点?

您可以使用

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']]

如果字典列表中没有重复的键,则合并这些键

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"]]
instrumets = ({700: "guitar"}, {701: "drums"})
instruments_dict = {}
for d in instrumets:
    instruments_dict.update(d)
for m in musicians:
    m[-1] = instruments_dict.get(m[0], m[-1]) # thanks to  Ev. Kounis 

最新更新