如何在Cython中正确实现类型转换，兼容Python参数输入？

问题背景 #

我编写了如下Cython函数，希望在Python中调用时，能像原生函数一样无需区分参数是否带小数点：

cdef double __test_func(double x, double y, double z):
    return (x-y)/((2*x-y)*y)**(0.5*z)
def test_func(x, y, z):
    return __test_func(<double>x, <double>y, <double>z)

我用<double>做类型强制转换，但不确定是否正确。后续尝试给Python包装器用融合类型Numeric：

ctypedef fused Numeric:
    char
    short
    int
    long
    long long
    float
    double
    long double
cdef double __test_func(double x, double y, double z):
    return (x-y)/((2*x-y)*y)**(0.5*z)
def test_func(Numeric x, Numeric y, Numeric z):
    return __test_func(<double>x, <double>y, <double>z)

编译后发现传入整数（如x=100）和浮点数（如x=100.）的结果差异极大，像是执行了C语言的强制截断。

问题原因 #

Cython中的<double>是C风格的强制类型转换，对于整数类型会直接按二进制位截断而非数值转换，这和Python原生的数值转换逻辑完全不同，因此导致结果异常。

解决方案 #

方案1：直接使用Python float类型转换（推荐）

在Python包装器中先将参数转为Python的float（对应C的double，64位双精度），再传递给内部Cython函数，完全对齐Python原生行为：

cdef double __test_func(double x, double y, double z):
    return (x-y)/((2*x-y)*y)**(0.5*z)

def test_func(x, y, z):
    # 转为Python float（即C的double），确保数值转换正确
    return __test_func(float(x), float(y), float(z))

方案2：融合类型下的安全转换

如果需要保留融合类型的静态编译优化优势，可通过数值运算触发安全的类型提升，避免C风格截断：

ctypedef fused Numeric:
    char
    short
    int
    long
    long long
    float
    double
    long double

cdef double __test_func(double x, double y, double z):
    return (x-y)/((2*x-y)*y)**(0.5*z)

cdef double _convert_to_double(Numeric val):
    # 通过乘以1.0触发数值提升，将整数/单精度浮点转为双精度
    return val * 1.0

def test_func(Numeric x, Numeric y, Numeric z):
    return __test_func(_convert_to_double(x), _convert_to_double(y), _convert_to_double(z))

关键说明 #

• Cython中float(x)调用的是Python的float类型，对应C标准的double（64位浮点数），精度和Python原生完全一致，不用担心精度损失。注意不要混淆Python的float和Cython中声明的C类型float（32位单精度）。
• 方案1无需额外处理融合类型，适合大多数场景；方案2则兼顾了静态类型优化和正确的数值转换逻辑。

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