嵌套列表Python 3中的数字平方

前端之家收集整理的这篇文章主要介绍了嵌套列表Python 3中的数字平方前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我必须解决的问题是将嵌套列表作为输入,并返回相同的嵌套列表,除了每个元素是以前存在于该点中的元素的平方.这是我的代码
>>> def treemap(lst):
...     for element in lst:
...         if element == type(list):
...             return treemap(element)
...         else:
...             element=element**2
...     return lst
>>> lst = [1,2,3,[4,[5,6],7]]
>>> print(treemap(lst))

现在我收到一个错误,说’int’对象不可迭代.我假设这意味着它正在尝试为整数类型运行循环,这对我来说没有意义,因为我只重新运行列表类型的函数.

解决方法

>不要在递归调用中返回,一旦返回,您将停止处理所有剩余元素
> element == type(list)不正确,因为type(list)是< class'type'>它永远不会等于列表中的任何项目.而是使用isinstance
>在基本情况下,您需要按索引访问元素以反映更改
def treemap(lst):
    for i,element in enumerate(lst):
        if isinstance(element,list):
            treemap(element)
        else:
            lst[i] = lst[i]**2
    return lst

输出

[1,4,9,[16,[25,36],49]]

猜你在找的Python相关文章