Cython中std::complex<double>向量转double complex内存视图及类型配置方法

C++标准明确规定std::complex<double>的内存布局和C语言原生的double complex完全一致：都是连续存储的两个double值，实部在前、虚部在后，因此可以直接通过指针强转的方式绑定内存视图，不需要做数据拷贝。
转换代码如下：

# 需保证vec在内存视图的整个生命周期内未被销毁、未发生扩容等改变首地址的操作
cdef complex[::1] mem_view = <complex[:vec.size()]> &vec[0]

注意：内存视图仅持有底层数据的引用，不管理std::vector的生命周期。如果需要脱离原vector独立使用数据，可调用np.asarray(mem_view)将数据拷贝到numpy数组中。

有两种常用配置方式，可根据使用场景选择：

• 单文件局部配置：在.pyx文件头部添加Cython编译器指令即可，当前文件内所有complex类型声明会默认使用std::complex<double>：

  # cython: language=c++
  # cython: ccomplex=libcpp.complex.complex[double]
  
  # 后续代码中直接使用complex即可，无需额外重定义
  from libcpp.vector cimport vector
  
  cdef vector[complex] vec
  

• 全项目全局配置：在setup.py的编译配置中添加编译器指令，所有被编译的Cython文件都会默认使用std::complex<double>作为complex的底层实现：

  from setuptools import setup, Extension
  from Cython.Build import cythonize
  
  ext = Extension(
      name="你的模块名",
      sources=["你的模块源码.pyx"],
      language="c++"
  )
  
  setup(
      ext_modules=cythonize(
          ext,
          compiler_directives={"ccomplex": "libcpp.complex.complex[double]"}
      )
  )
  

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