从文本文件python中创建嵌套列表的函数

Question

处理一个函数，该函数在块之间的空间上创建一行文本并将它们转换为基于 的内部列表。该代码适用于除最后一个块之外的所有块，它忽略了或某物。

代码：

def build_grocery_list(grocery_file): 
    food_list = [] 
    inner_list = [] 
    for line in grocery_file: 
     if line.strip() == "": 
      food_list.append(inner_list) 
      inner_list = []    
     if line.strip() != "": 
      inner_list.append(line.strip()) 

    return food_list 

回报：[['milk', 'cheese', 'cream cheese', 'eggs'], ['bread', 'buns', 'pita'], ['hamburgers', 'hotdogs', 'chicken'], ['chicken soup', 'canned tuna']]

缺少最后一块：lettuce, cabbage

Answer 1

如果你有文字的末尾空行文件，它会奏效。如果不是，它永远不会追加最后的inner_list。只需添加一个检查并在退出循环后追加。

def build_grocery_list(grocery_file): 
    food_list = [] 
    inner_list = [] 
    for line in grocery_file: 
     if line.strip() == "": 
      food_list.append(inner_list) 
      inner_list = []    
     if line.strip() != "": 
      inner_list.append(line.strip()) 
    if (inner_list): 
     food_list.append(inner_list) 

    return food_list