如何在Cython纯Python模式下扩展Python内置float类型？

问题背景 #

你需要将Python中继承自float的Bearing类转换为Cython代码，既要保留纯Python模式下的代码复用性，又要解决Cython扩展类型不支持__new__的问题，同时提升大量实例化和运算的性能（用于飞机轨迹模拟）。

可行方案 #

方案1：纯Python兼容的Cython优化版 #

直接基于原Python类结构添加Cython类型注解和优化装饰器，无需改动核心逻辑，解释型和编译型代码共用同一文件：

# bearing.pyx
import cython

@cython.typecheck
class Bearing(float):
    def __new__(cls, value: cython.double):
        value %= 360.0
        return super().__new__(cls, value)

    @cython.cfunc
    def __add__(self: 'Bearing', other: 'Bearing') -> 'Bearing':
        return Bearing(cython.cast(cython.double, self) + cython.cast(cython.double, other))

优化点说明：

• @cython.typecheck开启编译期类型检查，帮助Cython生成更高效的机器码
• @cython.cfunc将方法编译为C级别的函数，减少Python方法调用的开销
• 使用cython.double和cython.cast明确类型，避免隐式类型转换开销
• 完全兼容纯Python解释运行，直接执行bearing.pyx即可验证逻辑正确性

方案2：混合模式极致优化（解决加法性能瓶颈） #

如果需要进一步提升运算性能，可将实例化归一化逻辑与运算逻辑拆分：用Cython扩展类型处理高效运算，Python子类处理实例化时的取值归一化：

# bearing.pyx
import cython

# Cython扩展类型，专注高效运算
@cython.cclass
class _BearingBase(float):
    @cython.cfunc
    def __add__(self: '_BearingBase', other: '_BearingBase') -> 'Bearing':
        # 运算结果直接交给Bearing类做归一化
        return Bearing(cython.cast(cython.double, self) + cython.cast(cython.double, other))

# Python子类，负责实例化时的取值约束，继承扩展类型的高效方法
class Bearing(_BearingBase):
    def __new__(cls, value: cython.double):
        value %= 360.0
        return super().__new__(cls, value)

优势：

• 实例化时的归一化逻辑完全保留原Python逻辑，确保算法正确性
• 加法运算由Cython扩展类型处理，性能接近纯C运算水平
• 依然支持纯Python模式运行，扩展类型在解释时会自动退化为普通Python类

针对减法与BearingDelta的扩展建议 #

后续添加减法运算并返回BearingDelta类时，可复用相同模式：

• 为BearingDelta创建对应的Cython扩展基类，实现高效减法逻辑
• 用Python子类处理BearingDelta的实例化约束
• 在Bearing的减法方法中直接返回BearingDelta实例，确保运算性能

性能问题分析 #

你提到的之前方案的性能差异原因：

• 实例化速度快是因为跳过了部分Cython扩展类型的初始化开销，但运算时仍未充分利用类型优化
• 存储float的解释型类更快是因为避免了float子类的额外属性访问开销，但Cython编译后的扩展类型通过明确类型声明和cfunc装饰器，可弥补甚至反超这个差距

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