为多个变量赋值时如何迭代dict键

Question

我有一个多维字典，主键是散列，子值键入单个字符。我试图通过迭代包含我想要的子键字符的字符串来将值的子集分配给多个变量。比如：

A,B,C = tree[hash][i] for i in "xyz"

这相当于：

A = tree[hash]["x"]
B = tree[hash]["y"]
C = tree[hash]["z"]

但尝试在一行中完成所有操作，可能是嵌入一个函数，我将在其中传递变量列表和相应的子键字符串。

Answer 1

你几乎做到了。只需在右侧表达式周围添加方括号。它被称为列表理解：

A, B, C = [tree[hash][i] for i in "xyz"]

Answer 2

tree[hash][i] for i in "xyz" 已经是一个生成器推导，但它需要用括号括起来，除非它作为函数的唯一参数传递，如下所示：

my_function(tree[hash][i] for i in "xyz")   # this is valid
my_function(tree[hash][i] for i in "xyz", 12)   # this is invalid, parsing conflict
my_function((tree[hash][i] for i in "xyz"), 12)   # this is valid

这与您的赋值表达式相同。解析时需要括号以避免歧义。

this answer 解决了这个问题，但在生成器理解足够时创建了一个不必要的列表：

A, B, C = (tree[hash][i] for i in "xyz")

解包通过迭代生成器推导来分配左侧变量，而不创建不必要的列表。

也可以避免散列 3 次，使用变量

h = tree[hash]
A, B, C = (h[i] for i in "xyz")

带有map 和operator.itemgetter 的变体以避免循环：

import operator
A,B,C = map(operator.itemgetter(tree[hash]),"xyz")

另外：避免将hash 作为变量名，因为它是散列对象的内置函数。