python - Cython numpy 数组形状，元组赋值

Question

我用的成语

 size_x, size_y, size_z = some_array.shape

在处理 numpy 数组时很常见。当所讨论的数组具有类型时，同样的事情在 Cython 中似乎不起作用，例如

 def someFunc(np.ndarray[np.float32_t, ndim=2] arr):
      sx, sy = arr.shape

我们最终遇到了这样的编译错误

  Cannot convert 'npy_intp *' to Python object

这可能是因为“形状”被转换为 C 数组(为了更快的访问)，所以它不再是一个元组。

是否有可能在 Cython 中以某种方式提取这个元组？ (或者我应该坚持使用 sx, sy = arr.shape[0], arr.shape[1]？)

Answer 1

我相信你是正确的，处理这个问题的直接方法是这样的:

cdef int sx, sy
sx = arr.shape[0]
sy = arr.shape[1]

我不知道还有其他方法可以做到这一点，这是我在自己的代码中使用的约定。