python - 使用 C 将 numpy 数组对象拆分为两个不同大小的 vector

Question

我有

X 作为输入 --- 这是 dtype 对象这是以下结构x=[[1,2,3,4...n个元素],[1个元素],[1,2,...m个元素],[1个元素]]

模拟输入...

>>> from numpy import *
>>> x=array([array([1,2,3,4,5]),array([1]),array([1,2,3,4,5,6,7,8]),array([1])],dtype=object)
>>> x
array([[1 2 3 4 5], [1], [1 2 3 4 5 6 7 8], [1]], dtype=object)

我将 X 作为 PyArray_Object 的参数传递给我的 Python C 扩展

static PyObject* samp(PyObject *self, PyObject *args) {

    PyArrayObject *array,*p1,*p2;
    int n,j;

    if (!PyArg_ParseTuple(args, "O!",&PyArray_Type, &array))
        return NULL;
    n=array->nd;

    if(n!=1 || array->descr->type_num!=PyArray_OBJECT) {
        PyErr_SetString(PyExc_ValueError, "array must be one-dimensional and of Object type");
        return NULL;
    }
    j=array->dimensions[0];
    /* ...... */
}

现在我被困在这里，因为我不确定如何将它分成 4 个对象请任何人给我一些关于这个的指示......

Answer 1

最初您有一个包含四个数组的数组。以下行将其提取到四个数组变量中:

a,b,c,d=x[0],x[1],x[2],x[3]

如果您需要原生 Python 对象而不是 numpy 数组，请使用列表理解:

objs = [y for y in a]