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()
运行,输出为:
nested_listis:['a','b','c','d'] flatten_listis:['a','b','c','d']
nested_list方法采用递归的方式,如果item是list类型,继续递归调用自身。如果不是,将item加入结果列表中即可。
flatten_list方法则是采用生成器的方式,本质上也是递归的思路。
两层嵌套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)
运行
[1,2,3,4,5,6,7]原文来自:https://www.py.cn
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END
暂无评论内容