将所有字段从一个Python数据类实例复制到另一实例的最简单方法

时间:2019-09-16 18:56:07

标签: python python-dataclasses

假设您已经定义了Python数据类:

@dataclass
class Marker:
    a: float
    b: float = 1.0

将值从实例marker_a复制到另一个实例marker_b的最简单方法是什么?

以下是我尝试实现的示例:

marker_a = Marker(1.0, 2.0)
marker_b = Marker(11.0, 12.0)
# now some magic happens which you hopefully can fill in
print(marker_b)
# result: Marker(a=1.0, b=2.0)

作为边界条件,我不想创建新实例并将其分配给marker_b。好的,我可以遍历所有定义的字段,然后逐个复制值,但是我想必须有一种更简单的方法。

4 个答案:

答案 0 :(得分:4)

dataclasses.replace函数返回该对象的新副本。 在不传递任何更改的情况下,它将返回没有修改的副本:

>>> import dataclasses
>>> @dataclasses.dataclass
... class Dummy:
...     foo: int
...     bar: int
... 
>>> dummy = Dummy(1, 2)
>>> dummy_copy = dataclasses.replace(dummy)
>>> dummy_copy.foo = 5
>>> dummy
Dummy(foo=1, bar=2)
>>> dummy_copy
Dummy(foo=5, bar=2)

请注意,这是一个浅表副本。

答案 1 :(得分:3)

我认为遍历字段可能是最简单的方法。我能想到的所有其他选项都涉及创建一个新对象。

from dataclasses import fields

marker_a = Marker(5)
marker_b = Marker(0, 99)

for field in fields(Marker):
    setattr(marker_b, field.name, getattr(marker_a, field.name))

print(marker_b)  # Marker(a=5, b=1.0)

答案 2 :(得分:1)

@dataclass
class Marker:
    a: float
    b: float = 1.0

marker_a = Marker(0.5)

marker_b = Marker(**m1.__dict__)

marker_b

# Marker(a=0.5, b=1.0)

如果您不想创建新实例,请尝试以下操作:

marker_a = Marker(1.0, 2.0)
marker_b = Marker(11.0, 12.0)

marker_b.__dict__ = marker_a.__dict__.copy()

# result: Marker(a=1.0, b=2.0)

不确定这是否被认为是不好的破解...

答案 3 :(得分:0)

另一个可能更优雅的选项:

import dataclasses

marker_a = Marker(1.0, 2.0)
marker_b = Marker(**dataclasses.asdict(marker_a))