Question

我有一本字典，将人们与他们的技能联系起来：

d = {
    'A': {'name': 'Peter', 'skill': 'python'},
    'B': {'name': 'Mary', 'skill': 'c#'},
    'C': {'name': 'James', 'skill': 'java'}
 }

我有一个列表，其中指定了不同任务所需的技能：

e = [
    ['python', 'java'],
    ['c#', 'R']
]

我需要生成一个新列表，其中列出了具有所要求技能的人员的姓名（如果找不到，则为“ N / A”）。在这种情况下，结果将是：

rslt= [['Peter', 'James'], ['Mary', 'N/A']]

以下for循环会生成结果。

# with for loops
for r in range(len(e)):
    for s in range(0,len(e[r])):
        skill_found = 'false'
        for k in d.keys():
            if e[r][s] == d[k]['skill']:
                rslt[r][s] = d[k]['name']
                skill_found = 'true'
                break
        if skill_found == 'false':
                rslt[r][s] = 'N/A'
print(rslt)

但是，我真的很想获得一些列表/词典理解来做到这一点，但是我一直都失败了。以下两个语句生成垃圾：

#with list/dict comprehension
lstComp1 = [[d[k]['name'] if str(e[y][x]) == d[k]['skill'] else 'N/A' for x in range(0, len(e[y])) for k in d.keys()] for y in range(len(e))]
lstComp2 = [[x if x in d else 'N/A' for x in range(len(e[y]))] for y in range(len(e))]
print( lstComp1)
print(lstComp2)

我们将不胜感激。

Answer 1

如果您使用Python 3.8，则可以将for-loop转换为单行代码：

d = {
    'A': {'name': 'Peter', 'skill': 'python'},
    'B': {'name': 'Mary', 'skill': 'c#'},
    'C': {'name': 'James', 'skill': 'java'}
}

e = [
    ['python', 'java'],
    ['c#', 'R']
]

rslt = [[person['name'] if any((person:=v)['skill']==skill for v in d.values()) else 'N/A' for skill in task] for task in e]

print(rslt)

打印：

[['Peter', 'James'], ['Mary', 'N/A']]

Answer 2

@andrej的回答很好。我已经为任何python3.x

撰写了这篇文章

_any只是any的模仿，但在这里它返回一个字典以简化操作。

def _any(d,val):
    for v in d.values():
        if v['skill']==val:
            return v
    return {'name':'N/A'}
[[_any(d,v)['name'] for v in lst] for lst in e]
# [['Peter', 'James'], ['Mary', 'N/A']]

创建包含另一个列表中包含的字典值的列表

2 个答案: