python中的列表是可以嵌套的。将嵌套的list遍历并输出是很常见的需求。以下通过两种方法达到目的
defnested_list(list_raw,result): foriteminlist_raw: ifisinstance(item,list): nested_list(item,result) else: result.append(item) returnresult defflatten_list(nested): ifisinstance(nested,list): forsublistinnested: foriteminflatten_list(sublist): yielditem else: yieldnested defmain(): list_raw=["a",["b","c",["d"]]] result=[] print"nested_listis:",nested_list(list_raw,result) print"flatten_listis:",list(flatten_list(list_raw)) main()
让代码run起来,输出为:
nested_listis:['a','b','c','d'] flatten_listis:['a','b','c','d']
nested_list方法采用递归的方式,如果item是list类型,继续递归调用自身。如果不是,将item加入结果列表中即可。
flatten_list方法则是采用生成器的方式,本质上也是递归的思路。
推荐学习《python教程》
2.两层嵌套list去重
list里面套了一层list,需要去重,并在生成一个去重的list。请看代码:
defdup_remove_set(list_raw): result=set() forsublistinlist_raw: item=set(sublist) result=result.union(item) returnlist(result) defmain(): list_dup=[[1,2,3],[1,2,4,5],[5,6,7]] printdup_remove_set(list_dup)
让代码run起来:
[1,2,3,4,5,6,7]
基本思路:将每一个子list转为set,然后求并集,即可。
3.多重嵌套去重
defdup_remove(list_raw,result): foriteminlist_raw: ifisinstance(item,list): dup_remove(item,result) else: result.add(item) returnlist(result) defmain(): list_raw=["a",["b","c",["d","a","b"]]] result=set() print"dup_removeis:",dup_remove(list_raw,result)
让代码run起来:
dup_removeis:['a','c','b','d']原文来自:https://www.py.cn
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END
暂无评论内容