给定一个项目列表[音乐家]和一个词典列表[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