0

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

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)

非常感激任何的帮助。

4

2 回答 2

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']]
于 2020-05-04T18:36:56.797 回答
2

如果您使用 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']]
于 2020-05-04T18:22:38.497 回答