Python对列表或者字典进行复制
这是在阅读nn.DataParallel源码的时候看到的一个模块,可以复制你所传入的列表,或者字典,或者元组,代码逻辑涉及到递归和zip的使用,我着实没看懂,但是很是优雅,记录一下。
1 2 3 4 5 6 7 8 9 10 11 | def copy_obj( input ,number): targets_list = range (number) def scatter_map(obj): if isinstance (obj, tuple ) and len (obj) > 0 : return list ( zip ( * map (scatter_map, obj))) if isinstance (obj, list ) and len (obj) > 0 : return [ list (i) for i in zip ( * map (scatter_map, obj))] if isinstance (obj, dict ) and len (obj) > 0 : return [ type (obj)(i) for i in zip ( * map (scatter_map, obj.items()))] return [obj for _ in targets_list] return scatter_map( input ) |
其中最重要的函数是scatter_map(不要纠结名称),通过递归的方式生成和原来一摸一样的对象。最后copy_obj返回一个列表,其中每个元素就是被复制的对象,下面看看例子。
例子一:
1 2 | input = { '键1' : 112 , '键2' : '12' } print (copy_obj( input , 4 )) # 把input复制4次 |
输出:
例子二:
1 2 | input = ([ 1 , 2 , 3 ], "元组" ) print (copy_obj( input , 4 )) # 把input复制4次 |
输出:
以上内容如有错误,恳请指正
分类:
Python操作
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· SQL Server 2025 AI相关能力初探
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南