我正在尝试从包含笛卡尔坐标x,y和z的3个列表中计算出类似半径的数量。 下面是我的最小代码示例,用于重现我面临的问题; 子类计算半径数量,但返回零值。 这是什么原因,如何解决?
脚本:
# -*- coding: utf-8 -*-
from dataclasses import dataclass, field
from typing import List
@dataclass
class LoadHalo:
x: List = field(default_factory=list)
y: List = field(default_factory=list)
z: List = field(default_factory=list)
def __post_init__(self):
self.x = [1, 2, 3]
self.y = [1, 3, 5]
self.z = [1, 4, 7]
@dataclass
class BinHalo(LoadHalo):
r: List = field(default_factory=list)
def __post_init__(self):
self.r = self.modulus(self.x, self.y, self.z)
def modulus(self, *args):
"""Modulus of vector of arbitrary size."""
return sum([i ** 2 for containers in args for i in containers]) ** .5
halo = BinHalo()
print(f"halo.x: {halo.x}")
print(f"halo.r: {halo.r}")
输出 x 和 r 的以下值:
halo.x: []
halo.r: 0.0
因为您正在覆盖__post_init__
,所以列表仍然是空的,因为它们从未使用空列表默认值以外的任何内容进行初始化。如果还需要超类方法的行为,则必须调用要重写的超类方法。
您想要的是以下内容:
def __post_init__(self):
super().__post_init__()
self.r = self.modulus(self.x, self.y, self.z)
请注意类型提示的几件事:您可能希望为子类中的r
字段以及返回类型的modulus
float
:
def modulus(self, *args) -> float:
...
此外,您应该使用float
对象初始化默认列表,以便您可以编写:
x: List[float]
对于您的列表字段,因为您可能要使用浮点数学。
总而言之,我会定义所有内容:
from dataclasses import dataclass, field
from typing import List
@dataclass
class LoadHalo:
x: List[float] = field(default_factory=list)
y: List[float] = field(default_factory=list)
z: List[float] = field(default_factory=list)
def __post_init__(self) -> None:
self.x = [1.0, 2.0, 3.0]
self.y = [1.0, 3.0, 5.0]
self.z = [1.0, 4.0, 7.0]
@dataclass
class BinHalo(LoadHalo):
r: float = 0.0 # or whatever is suitable
def __post_init__(self) -> None:
super().__post_init__()
self.r = self.modulus(self.x, self.y, self.z)
def modulus(self, *args: List[float]) -> float:
"""Modulus of vector of arbitrary size."""
return sum([i ** 2 for containers in args for i in containers]) ** .5