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

Question

在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}

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() }

final if子句用作过滤器,与条件表达式不同.

Answer 2

如果您有不同的条件来评估键和值，@Marcin 的答案就是正确的选择。

如果键和值的条件相同，则最好在生成器表达式中构建 (key, value) 元组，并将其输入dict()：

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

它更容易阅读，并且每个键、值仅评估一次条件。

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

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

假设您只想在其中keys添加后缀，并且在这种情况下您希望将其替换为集合的长度。否则，键值对应保持不变。avaluevalue

dict((f"{k}_a", len(v)) if "a" in v else (k, v) for k, v in d.items())
# {'key1_a': 3, 'key2': {'bar', 'foo'}, 'key3': {'sad', 'so'}}

Answer 3

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

假设您有以下字典集

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 4

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

{_ for _ in iterable if True}