2012-04-24 17 views
1

これは簡単な質問ではありますが、初心者ではありませんでした。Python C-エクステンションに渡されるnumpy配列要素を変更するには?

私はnumpyのアレイを使用して次のPython C-extenstionを有する:共有ライブラリとしてコンパイルした後

#include <cmath> 
#include <Python.h> 
#include <iostream> 
#include "numpy/arrayobject.h" 

using namespace std; 
PyObject *func_calc(PyObject *self, PyObject *args) 
{ 
    PyObject * PO_clmn; 
    PyArrayObject * py_clmn; 
    if (!PyArg_ParseTuple(args, "O", &PO_clmn)) 
     return NULL; 
    py_clmn = (PyArrayObject*)PyArray_ContiguousFromObject(PO_clmn,PyArray_DOUBLE,1,1); 
    double *clmn = (double*)(py_clmn->data); 

    int i; 
    int N = py_clmn->dimensions[0]; 
    int flag_threadholds[N_threadholds]; 
    for (i=0; i<N; i++) 
    { 
     clmn[i]=1; 
    } 
    return Py_None; 
} 

static PyMethodDef exampleMethods[] = 
{ 
    { "calc", func_calc, METH_VARARGS }, 
    { NULL, NULL } 
} ; 

PyMODINIT_FUNC initcalc() 
{ 
    import_array(); 
    Py_InitModule("calc", exampleMethods); 
} 

を、私は、次の呼び出しが「1」の要素clmn配列を変更することができなかったが見つかりました:

import numpy 
from calc import calc 
clmn=numpy.zeros(10) 
calc(clmn) 
print clmn #[0,0... 

ありがとうございます!

答えて

1

渡すデータによっては、PyArray_ContiguousFromObjectの呼び出しによって元のオブジェクトが返される場合や、オブジェクトのコピーが返される場合があります。コピーが返された場合、コードは元のオブジェクトではなくそのコピーを変更しています。

関連する問題