Cython设置wraparound=False时NumPy数组内存视图赋值异常原因咨询

使用Cython编写代码为NumPy数组内存视图的每行最后一个元素赋值时，出现如下异常：

• 编译时设置wraparound = False，数组最后一行的最后一个元素无法被正确赋值；
• 设置wraparound = True时，所有行的最后一个元素均能正常赋值为预期值。

以下是可在Jupyter Notebook中运行的简化复现代码：

加载Cython扩展 #

%load_ext cython

设置wraparound = False的测试代码 #

%%cython

# cython: boundscheck = False
# cython: wraparound = False

import numpy as np
cimport numpy as np

def set_array(np.float32_t[:, :] buffer):
    cdef Py_ssize_t i = 0
    cdef Py_ssize_t n = buffer.shape[0]

    for i in range(n):
        # 为第i行最后一个元素赋值1
        buffer[i, -1] = 1.0
        
    print(np.asarray(buffer))

调用代码 #

import numpy as np

x = np.zeros((3, 4), dtype=np.float32)

set_array(x)

输出结果（wraparound=False） #

[[0. 0. 0. 1.]
 [0. 0. 0. 1.]
 [0. 0. 0. 0.]]

可见最后一行末尾元素未被赋值为1。

输出结果（wraparound=True） #

[[0. 0. 0. 1.]
 [0. 0. 0. 1.]
 [0. 0. 0. 1.]]

符合预期。

核心逻辑差异 #

wraparound参数的作用是控制Cython是否允许负索引自动转换为合法正索引：

• 当wraparound = False时，Cython完全禁用负索引的语义转换，此时buffer[i, -1]会被直接当作内存偏移量处理，指向当前行起始位置向前偏移1个元素的内存地址——这属于越界访问。
• 对于前n-1行，由于NumPy数组的内存是连续的，这个越界地址刚好对应到当前行的最后一个元素（下一行起始位置的前一个元素），所以赋值结果巧合符合预期；
• 但处理最后一行时，buffer[i, -1]指向的是数组内存范围之外的地址，赋值操作无法影响到目标元素，因此最后一行的最后一个元素保持初始值0。

另外，开启boundscheck = False会关闭数组边界检查，导致越界访问不会抛出错误，只会出现不符合预期的结果。

方法1：显式使用正索引（推荐） #

直接通过数组形状计算最后一列的正索引，无论wraparound设置如何都能正常工作：

%%cython

# cython: boundscheck = False
# cython: wraparound = False

import numpy as np
cimport numpy as np

def set_array(np.float32_t[:, :] buffer):
    cdef Py_ssize_t i = 0
    cdef Py_ssize_t n = buffer.shape[0]
    cdef Py_ssize_t last_col = buffer.shape[1] - 1  # 显式计算最后一列索引

    for i in range(n):
        buffer[i, last_col] = 1.0
        
    print(np.asarray(buffer))

方法2：保留负索引，开启wraparound #

如果需要继续使用负索引语法，保持wraparound = True即可，仅会带来极小的性能损耗，多数场景下可忽略。

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