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

Question

[Python 2.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}

Answer 1

您已经掌握了：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子句充当过滤器，这与具有条件表达式不同。

Answer 2

@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'}}

Answer 3

如果您具有不同]条件