前端之家收集整理的这篇文章主要介绍了
嵌套列表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]]
原文链接：https://www.f2er.com/python/186186.html
