在多级列表理解中是否需要中间列表

Question

这是一个具体的例子：

my_dict={k:int(encoded_value) 
         for (k,encoded_value) in 
             [encoded_key_value.split('=') for encoded_key_value in 
              many_encoded_key_values.split(',')]}

问题是关于内部列表[]，可以避免吗，例如：

# This will not parse
my_dict={k:int(encoded_value) 
         for (k,encoded_value) in 
             encoded_key_value.split('=') for encoded_key_value in 
             many_encoded_key_values.split(',')}

...，这是无效的语法：

NameError: name 'encoded_key_value' is not defined

样本数据：aa=1,bb=2,cc=3,dd=4,ee=-5

Answer 1

如前所述，生成器表达式将增强您的方法，避免创建内部列表。但是有一种更短的方法来获得所需的结果，使用re.findall() 函数：

result = {k:int(v) for k,v in re.findall(r'(\w+)=([^,]+)', many_encoded_key_values)}
print(result)

输出：

{'dd': 4, 'aa': 1, 'bb': 2, 'ee': -5, 'cc': 3}

---

替代方法是使用返回'callable_iterator' 实例的re.finditer() 函数：

result = {m.group(1):int(m.group(2)) for m in re.finditer(r'(\w+)=([^,]+)', many_encoded_key_values)}

Answer 2

您可以通过使用中间生成器表达式来避免创建中间列表：

my_dict={k:int(encoded_value)
         for (k,encoded_value) in
             (encoded_key_value.split('=') for encoded_key_value in
              many_encoded_key_values.split(','))}

语法上几乎是一样的；不是先生成中间列表然后使用元素，而是动态消耗元素。

---

让这个过于冗长，您可以使用由生成器组成的“数据管道”：

eq_statements = (item.strip() for item in many_encoded_key_values.split(','))
var_i = (var_i.split('=') for var_i in eq_statements)
my_dict = {var: int(i) for var, i in var_i}
print(my_dict)

（不幸的是，.split 没有返回生成器，因此考虑到节省空间，这并没有多大用处……对于处理大文件，这样的事情可能会派上用场。）

找到this answer，其中split 作为迭代器。以防万一……

Answer 3

FWIW，这是一种功能性方法：

def convert(s):
    k, v = s.split('=')
    return k, int(v)

d = dict(map(convert, data.split(',')))
print(d)

输出

{'aa': '1', 'bb': '2', 'cc': '3', 'dd': '4', 'ee': '-5'}

Answer 4

一个简单而紧凑的变体，非常接近您最初的尝试：

d = {v.strip(): int(i) for s in data.split(',') for v, i in (s.split('='),)}

唯一的额外“技巧”是将s.split('=') 包装在一个元组中（用括号括起来：(s.split('='),)），以便在同一个for 迭代中获得split 的两个元素。其余的都很简单。