python – PyArray_SIZE的正确用法是什么？

发布时间：2020-12-20 13:51:55 所属栏目：Python 来源：网络整理

导读：我是新手操作C中的numpy数组.这里有一段代码让我1)将1-D,64-bit-float numpy数组列表传递给C和2)循环遍历每个数组,打印其内容. #include stdio.h#include stdlib.h#include stddef.h#include "Python.h"#include "numpy/arrayobject.h"#include "list_of_nda

我是新手操作C中的numpy数组.这里有一段代码让我1)将1-D,64-bit-float numpy数组列表传递给C和2)循环遍历每个数组,打印其内容.

#include <stdio.h>
#include <stdlib.h>
#include <stddef.h>
#include "Python.h"
#include "numpy/arrayobject.h"
#include "list_of_ndarrays_lib.h"

void print_all(PyObject *list) {

    int i,j;
    PyArrayObject *arrayObj;
    double *arrayData;

    for (i = 0; i < PyObject_Length(list); i++) {

        arrayObj = (PyArrayObject *) PyList_GET_ITEM(list,i);

        arrayData = PyArray_DATA(arrayObj);
        for (j = 0; j < PyArray_SHAPE(arrayObj)[0]; j++) {
            printf("%f ",arrayData[j]);
        }
        printf("n");

    }

}

这段代码可以工作,但是PyArray_SHAPE(arrayObj)[0]并没有推广到n维数组,如果我用PyArray_SIZE(arrayObj)替换它,我会得到一个分段错误.

PyArray_SIZE我做错了什么？

解决方法

只需添加import_array();调用print_all()开头.它将初始化数组函数C-API.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!