在Python 2.7+中是否存在一种方法来制作如下内容?
{ something_if_true if condition else something_if_false for key, value in dict_.items() }
我知道你可以用“如果”做任何事情:
{ something_if_true for key, value in dict_.items() if condition}
在Python 2.7+中是否存在一种方法来制作如下内容?
{ something_if_true if condition else something_if_false for key, value in dict_.items() }
我知道你可以用“如果”做任何事情:
{ something_if_true for key, value in dict_.items() if condition}
当前回答
@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'}}
你想用d2的键替换d1中的键如果它们的值相同,你可以这样做
# 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'}}
其他回答
另一个在字典理解中使用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
@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'}}
你想用d2的键替换d1中的键如果它们的值相同,你可以这样做
# 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'}}
值得一提的是,If only语句把If放在最后:
{_ for _ in iterable if True}
如果您有不同的条件来计算键和值,@Marcin的答案是正确的方法。
如果你对键和值有相同的条件,你最好在生成器表达式中构建(key, value)元组,并将其输入dict():
dict((modify_k(k), modify_v(v)) if condition else (k, v) for k, v in dct.items())
它更容易阅读,并且条件只对每个键,value计算一次。
借用@Cleb的集合字典的例子:
d = {'key1': {'a', 'b', 'c'}, 'key2': {'foo', 'bar'}, 'key3': {'so', 'sad'}}
假设您希望仅为值为a的键添加后缀,并且在这种情况下希望用集合的长度替换该值。否则,键值对应该保持不变。
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'}}
你已经知道了:A if test else B是一个有效的Python表达式。你的字典理解的唯一问题是,一个表达式在字典理解的地方必须有两个表达式,由冒号分隔:
{ (some_key if condition else default_key):(something_if_true if condition
else something_if_false) for key, value in dict_.items() }
最后的if子句充当筛选器,这与使用条件表达式不同。