在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}
你已经知道了: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
子句充当过滤器,这与具有条件表达式不同。
@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'}}