在c扩展中创建numpy数组段错误

Question

我只是想在开始编写扩展之前先创建一个 numpy 数组。这是一个超级简单的程序：

#include <stdio.h>
#include <iostream>
#include "Python.h"
#include "numpy/npy_common.h"
#include "numpy/ndarrayobject.h"
#include "numpy/arrayobject.h"

int main(int argc, char * argv[])
{
    int n = 2;
    int nd = 1;
    npy_intp size = {1};
    PyObject* alpha = PyArray_SimpleNew(nd, &size, NPY_DOUBLE);
    return 0;
}

该程序在PyArray_SimpleNew调用时出现段错误，我不明白为什么。我试图遵循之前的一些问题（例如numpy array C api和C array to PyArray）。我究竟做错了什么？

Answer 1

的典型用法PyArray_SimpleNew是例如

int nd = 2;
npy_intp dims[] = {3,2};
PyObject *alpha = PyArray_SimpleNew(nd, dims, NPY_DOUBLE);

请注意， 的值nd不得超过 array 的元素数量dims[]。

另外：扩展必须调用 import_array()来设置 C API 的函数指针表：

该函数必须在使用 C-API 的模块的初始化部分中调用。它导入存储函数指针表的模块并将正确的变量指向它。

例如在 Cython 中：

import numpy as np
cimport numpy as np

np.import_array()  # so numpy's C API won't segfault

cdef make_array():
  cdef np.npy_intp element_count = 100
  return np.PyArray_SimpleNew(1, &element_count, np.NPY_DOUBLE)