具有多个条件和值重新分配的列表理解

Question

我有一个函数有几个条件（确切地说是5）模拟这个语句：

for n, i in enumerate(EPS_List_EPSV):
        if i == 'Very Satisfied':
            EPS_List_EPSV[n] = 1
        elif i == 'Satisfied':
            EPS_List_EPSV[n] = 2
        elif i == 'Dissatisfied':
            EPS_List_EPSV[n] = 3
        else:
            EPS_List_EPSV[n] = 4

我想知道是否有更多的pythonic方式来写这个，因为函数的长度是荒谬的。我理解列表理解处理一个变量，但有没有办法用列表理解来写这个？正确地写它是在逃避我。

Answer 1

使用字典和列表理解：

corr = {"Very Satisfied":1, "Satisfied":2, "Dissatisfied":3}
EPS_List_EPSV = [corr.get(el, 4) for el in EPS_List_EPSV]

例如：

EPS_List_EPSV = ["Very Satisfied", "Satisfied", "Satisfied", "Dissatisfied", "Other"]

使用上面的代码变成：[1, 2, 2, 3, 4]

Answer 2

使用dict：

d = {
    'Very Satisfied': 1,
    'Satisfied': 2,
    'Dissatisfied': 3    
}

EPS_List_EPSV = ['Very Satisfied', 'Something Else', 'Satisfied', 'Something Else', 'Dissatisfied'

for n, i in enumerate(EPS_List_EPSV):
    EPS_List_EPSV[n] = d.get(i, 4)

print(EPS_List_EPSV)
[1, 4, 2, 4, 3]

如果给出任何未定义的密钥，d.get将返回4。

Answer 3

如果我是，我会创建一个字典

dictio = {'Very Satisfied' : 1, 'Satisfied': 2, 'Dissatisfied' : 3, 'other' : 4}

然后你的代码应该是这样的：

for n, i in enumerate(EPS_List_EPSV):
    EPS_List_EPSV = dictio[i]

编辑：克里斯的答案更准确，因为它不会引发KeyError，以防ESP_List_EPSV包含一个不在字典中的单词