Answers:
您已经知道了: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
子句充当过滤器,这与具有条件表达式不同。
{(a if condition else b): value for key, value in dict.items()}
将工作。
@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'}}
d1
,d2
我得到AttributeError: 'dict_values' object has no attribute 'index'
如果您有不同的条件来评估键和值,@ Marcin的答案就是解决方法。
如果键和值的条件相同,则最好在生成器表达式中构建(键,值)元组并馈入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
,并且在这种情况下,您希望将其替换为集合的长度。否则,键值对应保持不变。a
value
value
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'}}
在字典理解中使用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
dict
是由key:value
元素组成的,您不是在dict
这里构建a 而是在其中构建aset
(请参阅setliteral)。