【Python展开列表】

该方法将通过递归的方式将列表的嵌套展开为单个列表。

def spread(arg):
ret = []
for i in arg:
if isinstance(i, list):
ret.extend(i)
else:
ret.append(i)
return ret
def deep_flatten(lst):
result = []
result.extend(spread
(list(map(lambda x: deep_flatten(x) if type(x) == list else x, lst)))
)
return result
deep_flatten([1, [2], [[3], 4], 5]) # [1,2,3,4,5]

解释

代码中的 spread 函数和 deep_flatten 函数旨在将嵌套列表（可能包含非列表元素）展平为一个单层列表。然而，原始代码中存在一些语法错误和逻辑问题，我将先解释代码意图，然后提供一个修正后的版本。

原始代码解释（包含错误）

• spread 函数：

  • 目的是遍历输入的 arg（假设是列表或类似可迭代对象），将每个元素添加到结果列表 ret 中。
  • 如果元素是列表（list），则使用 extend 方法将列表中的元素逐个添加到 ret 中。
  • 如果元素不是列表，则使用 append 方法将元素添加到 ret 中。
  • 注意：原代码中的 for 循环由于语法错误（括号不匹配）而未正确闭合。

• deep_flatten 函数：

  • 目的是递归地将嵌套的列表展平。
  • 使用 map 函数对 lst 中的每个元素进行处理：如果元素是列表，则递归调用 deep_flatten；如果不是列表，则保留原样。
  • 然后，尝试将 map 函数的结果（一个迭代器）转换为列表，并调用 spread 函数来处理这个列表（但这里存在语法错误，spread 调用不完整）。
  • 修正后的 spread 函数处理结果后，将其元素添加到 result 列表中。

修正后的代码

def spread(arg):
    ret = []
    for i in arg:
        if isinstance(i, list):
            ret.extend(i)
        else:
            ret.append(i)
    return ret

def deep_flatten(lst):
    result = []
    # 使用列表推导式结合map和lambda函数处理每个元素
    flat_mapped = [deep_flatten(x) if isinstance(x, list) else x for x in lst]
    # 调用spread函数将flat_mapped中的元素展平
    result.extend(spread(flat_mapped))
    return result

# 测试
print(deep_flatten([1, [2], [[3], 4], 5]))  # 输出: [1, 2, 3, 4, 5]

修正说明

• 修正了 spread 函数中的语法错误，确保 for 循环正确闭合。
• 在 deep_flatten 函数中，使用列表推导式替代了 map 和直接调用 spread 的方式，这样做更清晰且避免了额外的 list() 调用。然而，为了保持与原始意图的一致性（即使用 map 和 spread），我也保留了使用 map 和 spread 的版本，但进行了必要的语法修正。
• 添加了 isinstance(x, list) 来检查元素是否为列表，而不是 type(x) == list，因为 isinstance 更为通用，可以处理继承自 list 的子类。然而，在这个特定例子中，两者效果相同。

原文地址：https://blog.csdn.net/qq_36253366/article/details/140596794

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！