如何在字典理解中使用if / else?

时间:2012-02-25 09:04:46

标签: python dictionary dictionary-comprehension

在python2.7中存在以下任何方式:

{ something_if_true if condition else something_if_false for key, value in dict_.items() }

我知道你可以用'if'

制作任何东西
{ something_if_true for key, value in dict_.items() if condition}

6 个答案:

答案 0 :(得分:160)

你已经知道了:A if test else B是一个有效的python表达式。你所显示的dict理解的唯一问题是dict理解中表达式的位置必须有两个表达式,用冒号分隔:

{ (some_key if condition else default_key):(something_if_true if condition 
          else something_if_false) for key, value in dict_.items() }

最终if子句充当过滤器,与条件表达式不同。

答案 1 :(得分:7)

@Marcin's answer涵盖了所有内容,但万一有人想看一个实际的例子,我在下面添加两个:

我们假设你有以下词组

d = {'key1': {'a', 'b', 'c'}, 'key2': {'foo', 'bar'}, 'key3': {'so', 'sad'}}

并且您想要创建一个新词典,其键指示字符串'a'是否包含在值中,您可以使用

dout = {"a_in_values_of_{}".format(k) if 'a' in v else "a_not_in_values_of_{}".format(k): v for k, v in d.items()}

产生

{'a_in_values_of_key1': {'a', 'b', 'c'},
 'a_not_in_values_of_key2': {'bar', 'foo'},
 'a_not_in_values_of_key3': {'sad', 'so'}}

现在让我们假设您有两个这样的词典

d1 = {'bad_key1': {'a', 'b', 'c'}, 'bad_key2': {'foo', 'bar'}, 'bad_key3': {'so', 'sad'}}
d2 = {'good_key1': {'foo', 'bar', 'xyz'}, 'good_key2': {'a', 'b', 'c'}}

如果相应的值相同,您希望用d1的键替换d2中的键,

# here we assume that the values in d2 are unique
# Python 2
dout2 = {d2.keys()[d2.values().index(v1)] if v1 in d2.values() else k1: v1 for k1, v1 in d1.items()}

# Python 3
dout2 = {list(d2.keys())[list(d2.values()).index(v1)] if v1 in d2.values() else k1: v1 for k1, v1 in d1.items()}

给出了

{'bad_key2': {'bar', 'foo'},
 'bad_key3': {'sad', 'so'},
 'good_key2': {'a', 'b', 'c'}}

答案 2 :(得分:3)

如果您具有 不同个条件来评估键和值,@ Marcin的答案就是解决之道。

如果键和值具有 相同的条件,那么最好在生成器表达式中构建(键,值)元组并馈入{ {1}}:

dict()

它更易于阅读,并且每个键值仅对条件进行一次评估。

借用@Cleb的集合字典的示例:

dict((modify_k(k), modify_v(v)) if condition else (k, v) for k, v in dct.items())

假设您只想在d = {'key1': {'a', 'b', 'c'}, 'key2': {'foo', 'bar'}, 'key3': {'so', 'sad'}} 后加上keys后缀a,并且在这种情况下希望将value替换为集合的长度。否则,键值对应保持不变。

value

答案 3 :(得分:1)

还值得一提的是,If only 语句将 if 放在最后:

{_ for _ in iterable if True}

答案 4 :(得分:0)

在字典理解中使用if / else的另一个示例

我正在为自己的办公室工作而使用数据输入桌面应用程序,这种数据输入应用程序通常会从输入小部件中获取所有条目并将其转储到字典中以进行进一步的处理(例如验证或编辑)我们必须将所选数据从文件返回到条目小部件等。

使用传统编码的第一轮比赛(8行):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

a_dic, b_dic = {}, {}

for field, value in entries.items():
    if field == 'ther':
        for k,v in value.items():
            b_dic[k] = v
        a_dic[field] = b_dic
    else:
        a_dic[field] = value
    
print(a_dic)
“ {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

第二轮我尝试使用字典理解,但是循环仍然存在(6行):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

for field, value in entries.items():
    if field == 'ther':
        b_dic = {k:v for k,v in value.items()}
        a_dic[field] = b_dic
    else:
        a_dic[field] = value
    
print(a_dic)
“ {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

最后,使用单行字典理解语句(1行):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

a_dic = {field:{k:v for k,v in value.items()} if field == 'ther' 
        else value for field, value in entries.items()}
    
print(a_dic)
“ {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

我使用python 3.8.3

答案 5 :(得分:-1)

假设我想创建一个标记等级字典,该标记等级映射标记例如

 Marks      Grade
 0-40    -  F
 40-50   -  D
 50-60   -  C and so on 

为此,我可以将字典理解与以下所示的多个if条件一起使用:

 marks_dict = {x:"F" if x<40 else "D" if x<50 else "C+" if x<60 else "C" if x<70 
 else "B+" if x<80 else "A" if x<90 else "A+" if x<=100 else "NULL" for x in range(0,101)}

这是在字典理解中可以使用多个if else条件的方法,但是请确保最后使用final else块,否则将引发错误。

谢谢。