嵌套字典解析：太多值解壓

Question

我嘗試重寫下面的代碼使用字典解析工作，只是爲了好玩：

import itertools 

with open('foo.txt') as f: 
    entities = f.read().splitlines() 

parsed_entities = [] 
while entities: 
    props = itertools.takewhile(lambda n: n != 'EOM', entities) 
    entity = {p.split('=')[0]: p.split('=')[1] for p in props} 
    entities = entities[len(entity)+2:] # Delete and skip EOM & newline 
    parsed_entities.append(entity) 

我要替換這一行：

entity = {p.split('=')[0]: p.split('=')[1] for p in props} 

隨着更好看的字典理解，這可能看起來像：

entity = {key: value for p in props for key, value in p.split('=')} 

當我嘗試這樣做，我收到以下錯誤：

ValueError: too many values to unpack (expected 2)

我在做什麼錯？使用ipdb.pm()我看到p是name=yam，這很好，但是key和value未定義。因爲這需要p.split()調用的每個結果具有完全相同元素

for key, value in p.split('=') 

：

Answer 1

你不能做到這一點。相反，你只是有一個單一（字符串）元素序列，長度可變。

你不得不換p.split()到另一個迭代第一：

entity = {key: value for p in props for key, value in (p.split('='),)} 

所以現在不是：

['key', 'value'] 

你得到：

(['key', 'value'],) 

其迭代只是一次，提供兩個解包值。

但是，您可以使用可在此處調用的dict();它消耗的(key, value)雙直接可迭代：

entity = dict(p.split('=') for p in props) 

你也應該儘量避免讀取整個文件到內存中，你可以使用文件作爲迭代直接：

from itertools import takewhile 

parsed_entities = [] 
with open('foo.txt') as f: 
    cleaned = (l.rstrip('\n') for l in f) 
    while True: 
     props = takewhile(lambda n: n != 'EOM', cleaned) 
     parsed_entities.append(dict(p.split('=') for p in props)) 
     try: 
      next(cleaned) # consume line after EOM 
     except StopIteration: 
      # no more lines 
      break