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次

输出:

 

 

  

posted @   Circle_Wang  阅读(176)  评论(0编辑  收藏  举报
编辑推荐:
· 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代理技术深度解析与实战指南
点击右上角即可分享
微信分享提示