如何创建继承complex类的Python自定义类并支持额外参数初始化？

错误原因分析 #

• complex是Python的不可变内置类型，这类类型的实例创建逻辑在__new__方法中，而非__init__。直接重写__init__时，实例已通过父类__new__创建完成，而complex.__new__仅接受最多2个参数，因此传入3个参数会触发TypeError。
• complex的real和imag属性是只读属性，实例创建后无法修改，所以在__init__中尝试修改这些属性会抛出AttributeError。

解决方案 #

方案1：用组合代替继承（推荐） #

继承不可变内置类型容易受限于只读属性和创建逻辑，更稳妥的方式是将complex实例作为自定义类的内部属性，同时保存额外参数，并实现必要方法让类的行为和普通复数一致：

class MyComplex:
    def __init__(self, real, imag, imag2):
        self.complex_num = complex(real, imag + imag2)
        # 保存额外传入的虚部参数
        self.imag2 = imag2

    # 支持直接转换为complex类型
    def __complex__(self):
        return self.complex_num

    # 保持和complex一致的打印格式
    def __repr__(self):
        return repr(self.complex_num)

    def __str__(self):
        return str(self.complex_num)

    # 代理real和imag属性
    @property
    def real(self):
        return self.complex_num.real

    @property
    def imag(self):
        return self.complex_num.imag

    # 重载加法示例，支持复数运算
    def __add__(self, other):
        result = self.complex_num + complex(other)
        # 可根据需求自定义运算后的imag2值
        return MyComplex(result.real, result.imag, 0)

# 测试
my_complex = MyComplex(3.0, 4.0, 5.0)
print(my_complex)  # 输出: (3+9j)
print(my_complex.real)  # 输出: 3.0
print(my_complex.imag)  # 输出: 9.0
print(my_complex.imag2)  # 输出: 5.0

方案2：继承complex（不推荐，但可实现） #

如果一定要继承complex，需在__new__中处理参数，并将额外参数作为自定义属性添加到实例上（complex实例允许添加自定义属性，只要不修改内置只读属性）：

class MyComplex(complex):
    def __new__(cls, real, imag, imag2):
        # 合并虚部后创建complex实例
        instance = super().__new__(cls, real, imag + imag2)
        # 添加自定义属性保存额外参数
        instance.imag2 = imag2
        return instance

# 测试
my_complex = MyComplex(3.0, 4.0, 5.0)
print(my_complex)  # 输出: (3+9j)
print(my_complex.real)  # 输出: 3.0
print(my_complex.imag)  # 输出: 9.0
print(my_complex.imag2)  # 输出: 5.0

注意：这种方式继承的运算方法返回的是普通complex类型，而非MyComplex，若需保持类型一致，需重载所有运算魔法方法。

内容的提问来源于stack exchange，提问作者Seba Rossi